March 22, 2022, 11:15 p.m. | /u/curimeowcat

Computer Vision www.reddit.com

I am trying to understand the concept of "head" in

[**yolov5/models/yolov5x.yaml**](https://github.com/ultralytics/yolov5/blob/ecc2c7ba73e71211b192cba69e255afad92de67a/models/yolov5x.yaml#L28)

Line 28 in [ecc2c7b](https://github.com/ultralytics/yolov5/commit/ecc2c7ba73e71211b192cba69e255afad92de67a)

head: , it would be great if you can explain it.

Specifically,(1) How is it different from simply an output from a neural network?(2) Does each head must have its corresponding labels so that we can compute losses? E.g. we I add a head to output class labels, I can use cross-entropy to train these addition classifier layers.(3) If I add an additional head, it …

computervision head yolov5

Data Architect

@ University of Texas at Austin | Austin, TX

Data ETL Engineer

@ University of Texas at Austin | Austin, TX

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne

Senior Machine Learning Engineer (MLOps)

@ Promaton | Remote, Europe

Senior Business Intelligence Developer / Analyst

@ Transamerica | Work From Home, USA

Data Analyst (All Levels)

@ Noblis | Bethesda, MD, United States