YoloV3 pt 모델을 최적화 하는 방법
367
작성한 질문수 7
안녕하십니까 교수님 기존에 custom cata 253장을 사용해 80epochs 학습한뒤 만들 모델이 480MB 정도 되는데요 (개별이미지 크기는 500KB 내외입니다 ) 웹캠을 사용해 돌려보니 컴퓨터 사양 문제도 있겠지만 , 정말 너무 느립니다
향후 웹캠이 아닌 고화질 카메라를 사용할 예정이고, 해당 모델의 고도화도 필요한 상황이라 데이터셋을 3천장정도 확보 해두고 재학습시키려고 하는데
모델을 크기를 결정하는 가장 큰 요소는 무엇인지 궁금합니다 (참고로 클래스는 1개입니다)
모델을 가볍게 만들면서도 성능을 좋게 만드는게 너무 어렵습니다....
답변 1
0
안녕하십니까,
학습이 느리다는 건가요? 아님 inference가 느린 건가요?
inference가 느리면 GPU를 사용하신 건가요? Yolo는 GPU가 아니고 일반 CPU에서는 많이 느립니다. 확인 부탁드립니다.
0
GPU 를 사용했습니다. 학습은 Colab에서 돌려서 4시간정도 소요되었고 inference 자체가 많이 느립니다 향후 60FPS 정도 나오는 카메라를 사용하려고 하는데 모델의 경량화를 목표로 재 학습 시키려고 합니다 2~3000개의 데이터 셋, 1클래스 식별이 가능한 모델을 만들고자 하는데 개별이미지 크기를 500KB 보다 더 줄여야 할까요?
0
inference 속도를 높이려면
1. 상대적으로 작은 이미지로 학습하고 inference 수행
2. 좀 더 성능이 높은 GPU를 사용(V100 정도?)
3. Quantization 적용(detect 성능은 좀 떨어집니다). 해당 github에서 자료를 찾아보시면 ultralytics에 quantization 적용할 수 있습니다.
현재 사용하시는 GPU 모델은 무엇이고, 몇 FPS정도 나오는지요?
MMDetection 버전 이슈
0
52
2
강의 환경설정 질문
0
61
2
Custom Dataset에서의 polygon 정보 관련
0
113
3
cvat.ai 보안 수준이 궁금합니다
0
98
2
캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.
0
120
3
추론 결과의 Precision(또는 mAP) 평가 방법
0
94
2
mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.
0
68
2
runpod에서 google drive 연결 시 오류 발생
0
126
2
로드맵 선택
0
74
1
mmcv
0
65
2
Anchor box의 Positive 처리 위치
0
71
2
해당 강의 runpod 적용 후 에러 제보드립니다
0
95
2
run pod credit 관련 제보
0
125
2
mmdetection 2.x과 3.x 호환 관련 표기
0
89
2
mm_faster_rcnn_train_kitti.ipynb 실행 오류
0
114
3
질문 드립니다.
0
89
3
mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.
0
89
1
강사님께 수정을 제안드리고 싶은 것이 있습니다.
0
102
1
google automl efficientdet 다운로드 및 설치 오류
0
87
1
이상 탐지에 사용할 비전 기술 조언 부탁드립니다.
0
111
2
OpenCV 관련 질문드립니다.
0
87
2
mmcv 설치관련해서 문의드려요
0
357
3
강의 구성 관련해서 질문이 있습니다
1
141
2
모델 변환 성능 질문드립니다.
0
126
1