GPU 하드웨어 성능이 떨어져서 발생하는 문제이다.
따라서, cfg 파일을 아래와 같이 변경하면 오류를 해결할 수 있다.
[net]
batch=64
subdivisions=64
width=64
height=64
channels=3
momentum=0.9
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1
learning_rate=0.001
burn_in=1000
max_batches = 2200
policy=steps
steps=4800,5400
scales=.1,.1
'인공지능 > 이미지 인식' 카테고리의 다른 글
AlexeyAB/darknet Yolo v4, v3 and v2 for Windows - 학습 방법 (0) | 2022.02.20 |
---|---|
AlexeyAB/darknet Yolo v4, v3 and v2 for Windows - 이미지 인식 (0) | 2022.01.23 |
AlexeyAB/darknet Yolo v4, v3 and v2 for Windows - 개발 환경 (0) | 2022.01.23 |