Open ghost opened 6 years ago
I got a same problem as yours. If U find a resolution, could U please E-mail it to me? rangqiongliu@gmail.com
Delete a checkpoint file. and then, re-run.
If you want to train with 2 classes with weights initialized from VGG16, than you need to exclude the SSD layers and mark them as trainable, since they were trained for 21 classes.
Try to add those lines to your command:
--checkpoint_exclude_scopes=ssd_300_vgg/conv6,ssd_300_vgg/conv7,ssd_300_vgg/block8,ssd_300_vgg/block9,ssd_300_vgg/block10,ssd_300_vgg/block11,ssd_300_vgg/block4_box,ssd_300_vgg/block7_box,ssd_300_vgg/block8_box,ssd_300_vgg/block9_box,ssd_300_vgg/block10_box,ssd_300_vgg/block11_box \
--trainable_scopes=ssd_300_vgg/conv6,ssd_300_vgg/conv7,ssd_300_vgg/block8,ssd_300_vgg/block9,ssd_300_vgg/block10,ssd_300_vgg/block11,ssd_300_vgg/block4_box,ssd_300_vgg/block7_box,ssd_300_vgg/block8_box,ssd_300_vgg/block9_box,ssd_300_vgg/block10_box,ssd_300_vgg/block11_box \
@hongym7 Delete which one?
@CJJ-717 Checkpoint file that created by previous learning
i want to train this code for num_classes=2 . I write this code in command prompt:
but I got this error :