Closed pykeras closed 2 years ago
SOLUTION - may help others:
solved the problem using --num_classes 80
I don't have time to check source code, and tell why thats the case but that is how I solve my problem
should mention that, on prediction the model act as 1 class object detection ( exactly the way I labeled my dataset and need it)
python train.py \
--dataset ./dataset/train.record \
--classes ./dataset/labels.names \
--num_classes 80 \
--mode eager_tf --transfer darknet \
--batch_size 8 \
--epochs 10 \
--weights ./checkpoints/yolov3.tf \
--weights_num_classes 80
Hi everyone,
I'm trying to fine-tune the model on a custom dataset, with 1 class only using following command:
But getting error as follow (first try) :
then double checked my dataset and label.names, second try goes as follow:
can't understand where the 67 value came from.