search

pjreddie / darknet

Convolutional Neural Networks
http://pjreddie.com/darknet/
Other
25.84k stars 21.33k forks source link

Yolo v3 Tiny Mask #851

Open felixendres opened 6 years ago

felixendres commented 6 years ago

Hi, similar question to #561, should the last yolo layer here have masks 0,1,2?

twtygqyy commented 6 years ago

Same question, why the masks are 1,2,3 instead of 0,1,2?

senliuy commented 6 years ago

Same question, shouldn't it be 0,1,2?

zhangming8 commented 6 years ago

Same question, shouldn't it be 0,1,2?

TheMikeyR commented 6 years ago

yes it should be 0,1,2, but since the original weights are trained on 1,2,3, you should not change it before @pjreddie adds new weights trained with correct masks or unless you train it yourself from the bottom on a big dataset such as imagenet to get a good generalised model which can act as a starting point for custom datasets.