[ubuntu] YOLOV4파일을 tflite형식으로 변환하기(convert YOLOV4 to tflite)

yolov4의 가중치(weight)파일을 tensorflow의 tflite형식으로 변경이 가능하다.

먼저 방법을 알아보기전에 밑의 것들이 설치가 되어있어야 한다.

 

opencv-python==4.4.0(4.1.1.26 버전이상)

twobeach.tistory.com/category/OpenCV/%EC%84%A4%EC%B9%98

'OpenCV/설치' 카테고리의 글 목록

pip install lxml
pip install tqdm
pip install tensorflow==2.3.0rc0
#cuda가 있는경우 gpu버전 설치
pip install tensorflow-gpu==2.3.0rc0
pip install absl-py
pip install easydict
pip install matplotlib
pip install pillow

위의 것들이 설치가 완료가 되었다면 이제 변환해주는 .py 파일들을 받아야한다.

 

git clone https://github.com/hunglc007/tensorflow-yolov4-tflite

다운이 완료되면 해당 디렉토리에 yolov4.weights 파일을 복사해서 넣어준다.

가중치파일이 없으면 밑의 링크에서 다운받으면 된다.

drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT

yolov4.weights

다운이 완료되면 밑의 명령어를 순서대로 터미널 창에서 작성해주면 tflite파일이 생성이 된다.

#Save darknet weights to tfmodel that's needed for tflite conversion:
python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4-416 --input_size 416 --model yolov4 --framework tflite

#Convert to Yolov4 tflite version:
python convert_tflite.py --weights ./checkpoints/yolov4-416 --output ./checkpoints/yolov4-416.tflite

#Additionally to test the converted tflite model:
python detect.py --weights ./checkpoints/yolov4-416.tflite --size 416 --model yolov4 --image ./data/kite.jpg --framework tflite

순서대로 명령어를 입력하면 밑의 사진과 같은 파일이 만들어진다.

 

---

꼭  yolov4.weights라는 가중치 파일로만 하는것이 아니라 언제든지 학습 된 가중치 파일을 이용해서 변환이 가능하다.