728x90 반응형 SMALL YOLOv5 성능 높이기1 [Object Detection] YOLOv5 성능 높이기 YOLOv5 성능 높이기 Background image 넣기 : 배경 이미지는 탐지할 객체가 없는 데이터들을 의미한다. 배경 이미지를 넣어주면 False Positives (FP)가 줄어드는 효과를 볼 수 있다. 전체 학습 데이터 셋에서 배경 이미지의 비율을 0 ~ 10% 정도를 추천한다. COCO 데이터 셋은 1000개의 배경 이미지를 가지고 있다 (학습 데이터 셋의 1%) Pretrained weights 사용 : 작거나 중간 정도 사이즈의 데이터셋에 추천한다 (train.py시 --weights 인자값 넣기). Epoch : epoch는 300부터 시작해서 overfit이 발생하면 줄이고 발생하지 않으면 600 또는 1200으로 점점 증가시킨다. 이미지 사이즈를 640으로 설정하기 : COCO 데이.. 2023. 6. 15. 이전 1 다음 728x90 반응형 LIST
