인프런 커뮤니티 질문&답변

공부공부님의 프로필 이미지
공부공부

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

YoloV3 와 yolov5 전이학습 pt 파일 크기 차이

작성

·

1.2K

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등의 옵션을 부여한건지 다시 한번 확인 부탁드립니다. 

공부공부님의 프로필 이미지
공부공부

작성한 질문수

질문하기