Adding a new head to the YOLO11n model to detect very small objects

Discussion
,
21

Thank you for your inputs.

I will try YOLO11n customization for GhostConv. However, if you find any Yolo11n based GhostConv YAML file, please share.

Best Regards,
Venkat

22

Hello Venkat,

Thanks for following up. As mentioned previously, adapting modules like GhostConv and C3Ghost (often used in YOLOv8 contexts) to the YOLO11 architecture requires careful handling due to differences, particularly with blocks like C3k2 and C2PSA used in YOLO11.

We don’t have an official, pre-validated yolo11n-ghost.yaml prototype available. Creating custom architectures by swapping modules often involves meticulous adjustments to ensure layer compatibility, correct channel counts, and proper tensor shapes throughout the network, as indicated by the TypeError you encountered.

Debugging these custom YAML files typically involves verifying the arguments and expected input/output dimensions for each layer, ensuring they align correctly, especially at connection points like Concat. Comparing your modified YAML structure against the standard yolo11n.yaml might help identify where the dimension mismatch is occurring.

Good luck with your experimentation!