
[DeepSeg] YOLOv5 설치 및 MNIST 데이터셋으로 학습
·
Side Project/DeepSeg
이전 포스트에서 변환한 MNIST 데이터셋으로 YOLOv5 모델을 학습시킨다. [DeepSeg] MNIST 데이터셋 YOLO 형식으로 변환 — zerogod 코코딩딩 [DeepSeg] MNIST 데이터셋 YOLO 형식으로 변환숫자 이미지 전처리를 위해 OpenCV로 ROI를 추출하던 중, YOLOv를 이용해 숫자 객체를 탐지한 후 이를 MNIST 모델을 이용해 분류하는 것이 낫겠다고 생각했다. 이 포스트에서는 MNIST 데이터셋을 YOLO용zerogod-ai-dev.tistory.com 1. YOLOv5 설치git clone https://github.com/ultralytics/yolov5.gitcd yolov5pip install -r requirements.txt프로젝트 디렉터리에서 YOLOv5를 ..