Skip to content

SiliconValleyInternship-KimChOI/YOLOv5_for_weight

Repository files navigation

yolov5_for_weight

written by deep learning&video processing team,
🎅MinJeong Kim - https://github.com/kimminje0ng
👨‍💻DongHeon Lee - https://github.com/Dongheon97
🙎‍♀️SeungMi Lee - https://github.com/minewet


Contains code modified yolov5 to suit purpose of our project. when we trained/testesd recognition model, we cloned this repository.
( train_and_test.ipynb, test.ipynb )

  • best.pt: Trained model. Can get this file from train_and_test.ipynb.

  • detect.py: Face recognition in video with trained model.
    We add more features ㅡ Confidence thresholding & Calculating appearance timeline of each character.



Currently, the same content has been incorporated into backend/yolov5.

About

used to clone for AI training/testing

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published