yolov5를 사용해서 Image detection 학습진행.
각 코드에서 수정해야 할 부분들을 잘 체크해야 한다.
학습 전 - 파일추가
train, test 데이터 파일 추가
data - data.yaml 파일 추가
학습 전 - 코드 수정
train
학습 후 - 파일 확인
runs > train > exp
학습하면서 결과들에 대해서 확인할 수 있다.
weights에 best.pt와 last.pt를 사용해서 test 파일을 돌려볼 수 있다.
학습 후 detect 파일
detect 결과 파일 확인
requirements.txt를 pip install -r로 설치를 했었는데 문제가 발생했었다.
가상환경을 conda prompt로 만들어줬던 건데 pycharm에서 설치를 하니 오류가 났었다.
requirements를 설치할때도 conda prompt를 사용해서 설치하니 해결되었다.