yolov4-tf2
yolov4-tf2 copied to clipboard
运行出错了
大哥,,,帮帮忙!!!!!
报错位置是这里:
model.fit(data_generator(lines[:num_train], batch_size, input_shape, anchors, num_classes, mosaic=mosaic), steps_per_epoch=max(1, num_train//batch_size), validation_data=data_generator(lines[num_train:], batch_size, input_shape, anchors, num_classes, mosaic=False), validation_steps=max(1, num_val//batch_size), epochs=Freeze_epoch, initial_epoch=Init_epoch, max_queue_size=1, callbacks=[logging, checkpoint, reduce_lr, early_stopping] )
TypeError: int() argument must be a string, a bytes-like object or a number, not 'tuple'
我解决这个问题了,win10 tensorflow2.0.0gpu 把fit函数换成了fit_generator函数,就可以解决了
好的