bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码,可以用于训练自己的模型。
MIT License
787 stars 184 forks source link

anchor index的问题 #40

Closed happy2019-k closed 3 years ago

happy2019-k commented 3 years ago

yolo脚本中第110行的代码:self.anchors_mask = [[3,4,5],[1,2,3]]是不是有问题,是否应该改为self.anchors_mask = [[3,4,5],[0,1,2]]。相同的问题:在nets/yolo_training.py中231和359行的anchor_index = [[3,4,5],[1,2,3]][self.feature_length.index(in_w)]是不是应该改成anchor_index = [[3,4,5],[0,1,2]][self.feature_length.index(in_w)]?