search

Chris-hughes10 / Yolov7-training

A clean, modular implementation of the Yolov7 model family, which uses the official pretrained weights, with utilities for training the model on custom (non-COCO) tasks.
GNU General Public License v3.0
116 stars 35 forks source link

Is there a bug in get_yolov7_e6e_config #15

Closed HexCrafts closed 1 year ago

HexCrafts commented 1 year ago

line 1152 - 1157 [ [257, 258, 259, 260, 261, 262, 263, 264], 1, detection_head, [num_channels, anchor_sizes_per_layer, strides, True], ],

does it should be the following? [ [257, 258, 259, 260, 261, 262, 263, 264], 1, detection_head, [num_classes, anchor_sizes_per_layer, strides, True], ],

Chris-hughes10 commented 1 year ago

Thanks for pointing that out, it has been fixed now.