Yolo4 가중치 Tf 변환 오류
포스트
취소

Yolo4 가중치 Tf 변환 오류

1
ValueError: cannot reshape array of size 379318 into shape (256,256,3,3)

직접 학습된 가중치 파일의 클래스 크기가 기본 yolo4와 맞지 않아서 발생

data 폴더에 자신의 class.names 파일을 넣고

core/config.py 의 class 부분에 자기 class 이름을 넣기

참고

https://github.com/hunglc007/tensorflow-yolov4-tflite/issues/126

이 기사는 저작권자의 CC BY 4.0 라이센스를 따릅니다.