Yolov5_model_maker
Yolov5 모델을 만드는 코드입니다. 코드의 의미는 주피터 노트북 파일에 각주로 자세히 달아놓았습니다. 개발 환경은 구글 코랩입니다. 코드 실행 시 데이터셋과 같은 파일의 경로가 올바르게 지정돼있는지 확인해주십시오.
dataset.zip에는 실제 인공지능 모델을 훈련할 때 사용했던 이미지 파일과 txt 라벨 파일, xml 어노테이션 파일이 있습니다. 이미 640 * 480으로 resize돼있기 때문에, 본 데이터셋을 그대로 쓸 것이라면 올려드린 코드에서 데이터셋 생성 관련 코드는 실행하지 않으셔도 됩니다.
모바일 클라이언트 관련 이해는 https://www.notion.so/b8e553f9b83f44b9a9d8663bba68608d 의 요구사항 3번을 참고하시기 바랍니다. 모델 동작을 위한 코드는 Yolov5 공식 github를 참고하세요.