yolo5 학습 진행 중 문의가 있어 이렇게 글 올립니다.
ex)
- 침대 클래스 학습 완료
- 침대 클래스 학습 완료된 파일에 의자 클래스 학습하여 침대+의자 학습 완료
- 침대+의자 학습 완료된 파일에 책상 학습을 하여 침대+의자+책상이 학습 완료된 데이터를 얻고 싶습니다.
침대 클래스 학습
python3 train.py --img 640 --batch 16 --epochs 30 --data ./dataset_cate/data_bed.yaml --cfg ./models/yolov5l.yaml --weights yolov5l.pt --name yolo5_bed
침대 클래스 학습 후 의자 클래스 학습
python3 train.py --img 640 --batch 16 --epochs 30 --data ./dataset_cate/data_bed_chair.yaml --cfg ./models/yolov5l.yaml --weights ./runs/train/yolo5_bed/weights/last.pt --name yolo5_bed_chair
침대, 의자 클래스 학습 후 책상 클래스 학습
python3 train.py --img 640 --batch 16 --epochs 30 --data ./dataset_cate/data_bed_chair_desk.yaml --cfg ./models/yolov5l.yaml --weights ./runs/train/yolo5_bed_chair/weights/last.pt --name yolo5_bed_chair_desk
라고 질문을 올렸던 분의 질문에 대한 답변에서 위와 같이 명령어를 작성해주셨는데 저는 침대 클래스가 학습이 안된 상태라고 가정한다면 제일 처음 침대 클래스 학습에서 --weight 옵션은 첫 학습에서 지워두는게 맞을까요?
ex)
python3 train.py --img 640 --batch 16 --epochs 30 --data ./dataset_cate/data_bed.yaml --cfg ./models/yolov5l.yaml --name yolo1_bed
아니면
ex)
python3 train.py --img 640 --batch 16 --epochs 30 --data ./dataset_cate/data_bed.yaml --cfg ./models/yolov5l.yaml --weights '' --name yolo1_bed
이렇게 하는게 맞는걸까요?
이전 질문에 대한 답변에는 --weights yolov5l.pt 라고 옵션에 적어주셨는데 해당 가중치 파일은 기존 질문자의 침대 클래스 학습 완료된 가중치 파일을 뜻하는걸까요?
또한 이미지가 제각각인 상태인 264개의 파일로 처음 학습한다고 가정하면 epochs 의 숫자와 데이터 모델의 사이즈의 연관관계는 어느 비율로 해주면 좋을까요?