A fellow learner reached out and asked me about my YOLO implementation. Some of the code is a mess right now because I was trying to break it apart in order to be able to train the classification separately from localization (as suggested in one of the YOLO papers). The classification layers are a…

Defining a custom class for YOLO loss

ai_curious January 21, 2022, 2:01pm 2

@big-bbox hope this helps. @ai2ys not sure if this exactly addresses your question, but maybe food for thought.

1 Like

Topic		Replies	Views
YoloV5 Output/ Model Not Learning with Custom Data Convolutional Neural Networks	2	496	May 28, 2023
How is the training done from 1919425 as labels are just class and boxes Convolutional Neural Networks	1	529	September 15, 2021
Instrumenting YOLO Training with TensorBoard Convolutional Neural Networks	1	595	June 10, 2022
C4W3 Yolo Algorithm Convolutional Neural Networks	4	613	June 25, 2021
Is YOLO a regression or classification algorithm? Convolutional Neural Networks	11	1018	February 23, 2023