시스템 시퀀스
- 데이터 크롤링 진행
- 크롬 버전에 맞는 webdriver 필요
$ cd utils; python crawl.py
-
메인보드 이미지에 대해 레이블링 진행
- CVAT에서 Annotation 처리
- yolo 포맷으로 저장
-
CPU, GPU, RAM에 대한 이미지 분류 학습 진행
$ cd Classification; python train.py
- Mainboard의 객체 검출 학습 진행
$ cd yolov5; python train.py --img 320 --batch 32 --epochs 100 --data coco.yaml --weights yolov5s6.pt
- 3, 4 단계에서 추출된 weight파일을 바탕으로 main.py에서 특정 부품 위치 검출 진행
$ python main.py