I don’t believe you could use the model as one “layer” in another model, as I would assume that the YOLOv5 model expects image tensor inputs and returns the predictions (not any intermediate activations), so I’m unsure how it should be used as a layer in a larger model.



 almog:

However, How do I actually train it?


This train.py 25 script from the repository seems to show the training of the model.