728x90
반응형
SMALL
YOLOv5 이어서 학습하기 --resume 사용
·
ML & DL
YOLOv5를 학습하다가 예기지 못하게 중간에 중단 현상이 나타났다.(사실 전적으로 본인의 잘못😭) 그래서 YOLOv5 학습 도중 중단되었을 때 학습을 이어나가는 방법을 정리하고자 한다.  YOLOv5 이어서 학습하기--resume을 사용하면 된다!  YOLO 처럼 유명한 모델은 대부분 weight를 기록하고 저장하는 코드가 굉장히 잘 짜여 있다. 그렇기 때문에 그런 모델을 사용할 경우에 우리는 있는 것을 잘 찾아서 사용하기만 하면 된다! 원래 다음과 같은 명령어를 실행해서 epoch는 100으로 설정하고  훈련시키고 있었다.python train.py --img 640 --batch 64 --epochs 100 --data data/data.yaml --cfg models/yolov5s.yaml --..
[YOLOv5] Custom Data Augmentation 하기 (roboflow)
·
ML & DL
object detection을 위한 데이터셋을 구축할 일이 있어서 찾아보다가 roboflow라는 사이트를 알게 되었다. vision task를 위한 여러 데이터셋이 이미 잘 구축되어 있고 라벨링도 본인이 원하는 방식에 맞게 다운받을 수 있어 알아두면 편할 것 같은 사이트이다!  다음 링크에 들어가면 많은 데이터셋 list가 나열되어 있다. https://public.roboflow.com/ Computer Vision DatasetsDownload free, open source datasets for computer vision machine learning models in a variety of formats.public.roboflow.com Data augmentation 하는 방법본론으로 들..
728x90
반응형
LIST
성장중 •͈ᴗ•͈