yolo-v4-tf.keras icon indicating copy to clipboard operation
yolo-v4-tf.keras copied to clipboard

Model Fails to load darknet weights when using custom classes.

Open nasheedyasin opened this issue 4 years ago • 2 comments

Following is the message I recieve: failed to read all weights, # of unread weights: 0 load from ./darknet_base_model/yolov4.weights

I am training it on the ASL (American Sign Language) dataset and it has 26 classes from A-Z. The class file is formatted as per standard (1 class per line).

nasheedyasin avatar Jan 13 '21 17:01 nasheedyasin

Is it necessary to use the pretrained weights in your case? The dataset you are using looks quite different from VOC/COCO anyway, so maybe you should just train the network from scratch.

gajdosech2 avatar Feb 21 '21 23:02 gajdosech2

They are different no doubt, but certain low level features that would be captured at the upper levels of the model would be fairly similar and I would ideally look to having those layers frozen, leaving the deeper layers, which more dataset specific features, unfrozen

nasheedyasin avatar Jul 01 '21 13:07 nasheedyasin