• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

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 전이학습

답변 1

답변을 작성해보세요.

0

안녕하십니까, 

일반적으로 Yolo v5가 v3보다 최적화가 잘되어서 모델의 사이즈가 좀 작습니다. 하지만 말씀하신만큼 작아지지는 않습니다(대략 70% 수준인것 같습니다). 

모델의 크기는 이미지의 갯수가 아니라 이미지의 크기에 영향을 받습니다. 두 버전의 이미지 크기를 다시 확인해 주시고, 혹 Yolo v5의 경우 quantization등의 옵션을 부여한건지 다시 한번 확인 부탁드립니다.