서버: 구글 Colaborator
데이터셋: https://public.roboflow.com/
이걸 복사해서 붙여넣고 실행시키면
이렇게 다운받아지고 압축 풀어짐
그리고 이 프로젝트에 yolov5를 클론받는다.
https://github.com/ultralytics/yolov5.git
그리고 cd yolov5로 그 디렉토리 안으로 이동한다.
그리고 위와 같이 pip install -r requirements.txt로 yolov5를 위한 패키지를 설치한다.
data.yaml에서 training경로와 validation경로가
*nc는 클래스 개수, names는 클래스 이름
glob을 통해 다운받은 모든 이미지들을 가져온다. 이미지 개수 출력해보니 2971개.
sklearn을 통해 트레이닝셋과 밸리데이션셋(테스트셋)으로 나눠준다.
이에 맞게 data.yaml을 변경해준다.
*yaml import 하는데 ImportError: No module named yaml 에러
=> pip install pyyaml 로 해결
Software Developer