search

WongKinYiu / YOLO

An MIT rewrite of YOLOv9
MIT License
614 stars 65 forks source link

RepNBottleneck -> RepBottleneck? #49

Closed TatsuyaShirakawa closed 3 months ago

TatsuyaShirakawa commented 3 months ago

Thanks for your great work!

Issue Description I don't fully understand the naming convention, but the current RepNBottlenck does not support repeat_num. Therefore, I think it would be better to rename it to RepBottleneck, similar to RepConv. https://github.com/WongKinYiu/YOLO/blob/ba6baa6c81bf198847d53a95d0fa9902fec69cb2/yolo/model/module.py#L201

henrytsui000 commented 3 months ago

Hi,

Got it. I think this is a mistake that I didn't catch during the code review of the module pull request. In the upcoming update, I will rename the class as you mentioned above.

Thank you for bringing this to my attention!

Best regards, Henry Tsui