아나콘다 파워셀 열고
conda activate p37_yolov5
작업 폴더로 이동
cd E:/yolo
비주얼 스튜디오 코드에서
파일 → 파일 열기 → yolo\models\yolov5s.yaml
yolo\data\Mask\data.yaml
두 파일의 nc 값이 일치해야 함
따라서 yolo\models\yolov5s.yaml를 백업본을 만든 뒤
yolo\models\yolov5s.yaml 파일의 nc 값을 2로 변경해준다
고친 다음
아나콘다 파워셀에서 학습 시켜준다
python train.py --img 416 --batch 16 --epochs 50 --data ./data/mask/data.yaml --cfg ./models/yolov5s.yaml --weights yolov5s.pt --name mask_yolov5s_results
