How to train with only 2 classes?

sfzhang15 / RefineDet

Single-Shot Refinement Neural Network for Object Detection, CVPR, 2018

Other

1.43k stars 393 forks source link

How to train with only 2 classes? #125

Closed YTLT closed 5 years ago

YTLT commented 5 years ago

Hi, I tried to train a network that only detects person. So I deleted other categories in labelmap_voc.prototxt. When I created new LMBD files(I run create_data.sh), there was an error: Unkown name: cow.

peterpaniff commented 5 years ago

Hi, I tried to train a network that only detects person. So I deleted other categories in labelmap_voc.prototxt. When I created new LMBD files(I run create_data.sh), there was an error: Unkown name: cow.

你需要将对应的标注文档xml也给改掉，你的标注xml文档只能出现person

YTLT commented 5 years ago

@peterpaniff I modified the xml files and it worked! thank you!

yisiyinheluojiutian commented 5 years ago

@peterpaniff I modified the xml files and it worked! thank you!

你好，请问具体需要怎么改啊？