종이주소 : https://arxiv.org/abs/2207.02696
출처 주소 : https://github.com/WongKinYiu/yolov7
1. 데이터세트
YOLOV5 데이터 세트를 루트 디렉터리에 직접 복사
이미지 폴더에는 사진이 포함되어 있고 레이블 폴더에는 해당 레이블이 포함되어 있습니다.
2. 미리 훈련된 가중치 다운로드
해당 모델은 해당 사전 훈련 가중치를 다운로드하여 루트 디렉토리에 배치합니다.
3. yaml 파일 구성
yolov7/data 폴더 아래에 새 VOC.yaml 파일을 생성합니다(COCO.yaml을 직접 복사).
1. 데이터 세트 주소를 해당 주소로 변경
2. 데이터셋의 카테고리 개수 및 카테고리 이름 변경
4. train.py 수정
.
epochs: 학습 라운드 수
batch_size: YOLOV7은 비디오 메모리를 많이 차지하므로 2 또는 4로 설정하는 것이 좋습니다.