Closed LIHANG-HONG closed 3 years ago
Hi, I have solved the problem by changing the order of the code. from
train_data_set = yolo.load_dataset(
'2007_train.txt',
image_path_prefix='VOCdevkit/VOC2007/JPEGImages',
label_smoothing=0.02
)
yolo.input_size=416
yolo.batch_size = 1
to
yolo.input_size=416
yolo.batch_size = 1
train_data_set = yolo.load_dataset(
'2007_train.txt',
image_path_prefix='VOCdevkit/VOC2007/JPEGImages',
label_smoothing=0.02
)
but I still dont know why.....
yolo.load_dataset
internally uses yolo.input_size
and yolo.batch_size
.
Hi, thank you for your great library. I met InvalidArgumentError using yolo4 to train a custom model. Here is my code.
here is the error I met when executing yolo.fit() on jupyter
the version is as below: yolov4 2.1.0 tensorflow 2.4.1 Keras 2.4.3
Could you please help me figure out the reason of this error?