-
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
YoloV3 와 yolov5 전이학습 pt 파일 크기 차이
22.01.13 18:30 작성 조회수 1.08k
0
안녕하십니까 교수님
yolov3. 253 장의 데이터를 사용해 학습하고
yolov5. 는 V3에 사용했던 253장과 추가 데이터 총 1839 장의 데이터를 사용해 학습을 진행했는데요
V3, V5성능차이가 있기때문에 학습시간의 차이가 있는것은 알겠으나 PT파일의 용량차이가 너무 큽니다
V3의 경우 481MB V5 의 경우 14 MB 인데요
훨씬 많은 데이터를 사용했고 동일하게 80 epoch씩을 학습시켰는데 이렇게 버전에 따라 모델의 용량차이가 극명하게 나는것이 맞는건가요?
v5 전이학습
V3 전이학습
답변을 작성해보세요.
0
권 철민
지식공유자2022.01.14
안녕하십니까,
일반적으로 Yolo v5가 v3보다 최적화가 잘되어서 모델의 사이즈가 좀 작습니다. 하지만 말씀하신만큼 작아지지는 않습니다(대략 70% 수준인것 같습니다).
모델의 크기는 이미지의 갯수가 아니라 이미지의 크기에 영향을 받습니다. 두 버전의 이미지 크기를 다시 확인해 주시고, 혹 Yolo v5의 경우 quantization등의 옵션을 부여한건지 다시 한번 확인 부탁드립니다.
답변 1