search

AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.6k stars 7.95k forks source link

Train with VGG #2608

Open weiting0032 opened 5 years ago

weiting0032 commented 5 years ago

@AlexeyAB VGG network is used on image classification, can it be used for image detection like yolo? If it can be used for image detection how do I edit the vgg16.cfg to make it works to detect? Thanks!

AlexeyAB commented 5 years ago

@WEITINGLIN32 Hi,

  1. Change these lines: https://github.com/AlexeyAB/darknet/blob/7a854302efb7adba80d5e8a747ad5e5ec384a226/cfg/vgg-16.cfg#L1-L17 to these: https://github.com/AlexeyAB/darknet/blob/7a854302efb7adba80d5e8a747ad5e5ec384a226/cfg/yolov3-tiny.cfg#L1-L23
  1. Change these lines: https://github.com/AlexeyAB/darknet/blob/7a854302efb7adba80d5e8a747ad5e5ec384a226/cfg/vgg-16.cfg#L130-L152 to these (but remove mask = 0,1,2 ): https://github.com/AlexeyAB/darknet/blob/7a854302efb7adba80d5e8a747ad5e5ec384a226/cfg/yolov3-tiny.cfg#L167-L182
weiting0032 commented 5 years ago

Hello, @AlexeyAB We don't need to remove mask = 3,4,5 in yolov3-tiny.cfg? Could you give me the edited cfg for me? Thanks!