search

choasup / caffe-yolo9000

Caffe for YOLOv2 & YOLO9000
Other
166 stars 134 forks source link

problems training 2-class detector #2

Closed vvvv3ummer closed 7 years ago

vvvv3ummer commented 7 years ago

Hi , i want to train a 2-class detector on caffe-yolo9000 , but i can't use the pretrained caffemodel that you provided cause it was trained on VOC which has 20 classes , otherwise it will report the error "Cannot copy param 0 weights from layer 'conv_reg'; shape mismatch. Source param shape is 125 1024 1 1 (12 8000); target param shape is 30 1024 1 1 (30720)."Is there some way to convert the yolo v2 weights into caffemodel , or how can i trained without pretrained model? Thanks a lot!

choasup commented 7 years ago

'conv_reg' layer in your train_prototxt should rename.