mathieugalle / pytorch-yolo-v1 Goto Github PK

Juste refaire le premier yolo pour être SÛR de bien comprendre le object detection.

Inspiré, le moins possible, mais quand même, par :

YOLOv1 à partir de zéro Aladdin Persson https://www.youtube.com/watch?v=n9_XyCGr-MI

et le papier : https://arxiv.org/pdf/1506.02640.pdf

Des Batch-Normalisations ont été rajoutées par rapport au papier parce qu'on est en 2022. J'ai utilisé pytorch parce que j'en ai plein le cul de Tensorflow.

La partie qui reste quand même un peu compliquée pour moi reste la définition du loss :-/

Je mets sur github juste pour les recruteurs (coucou), parce que sinon, ce code a peu d'intérêt.