can you train yolov5？

[WangPeng19981204]

i can not train yolov5 with your code ,report error ' ValueError: Target size (torch.Size([16, 1200, 1])) must be the same as input size (torch.Size([16, 17000, 1]))'，it seems that have prombles on network ,i find two prombles,the first is the neck of yolov5 is none ,the neck code need to write,the second is anchor size of yolov5 is that of yolov3 ,which need to be modified. finally ,thank you for you great work ,Look forward to your reply！

[caixiiaoyang]

i can not train yolov5 with your code ,report error ' ValueError: Target size (torch.Size([16, 1200, 1])) must be the same as input size (torch.Size([16, 17000, 1]))'，it seems that have prombles on network ,i find two prombles,the first is the neck of yolov5 is none ,the neck code need to write,the second is anchor size of yolov5 is that of yolov3 ,which need to be modified. finally ,thank you for you great work ,Look forward to your reply！

I have the same problem

[woolpeeker]

mmdetection-mini/mmdet/models/dense_heads/rr_yolov5_head.py

Line 73 in 3858c8c

nn.Conv2d(make_div8_fun(256), 255, 1),

I think the reason is that the yolov5Head‘s output channel number is hard coded.

[fengqian-wei]

I have the same problem. How to fix it? Thanks !!

[fengqian-wei]

I have the same problem. How to fix it? Thanks !!

setting as
nn.Conv2d(make_div8_fun(256), (classNum+5)*3, 1),