Yolov3 (tf1)를 tensorflow 2버전으로 실행해 보자

직접 만드는 것은 아니고 github에 있는 코드를 실행해 보자.

 

[21.03.03]  yolov3 깃허브 검색 조회 결과

• 쓸만한 모듈을 찾을 때는 업데이트 날짜도 잘 살펴보자. 특히나 기술 업데이트가 휙휙 발생하는 딥러닝의 경우 갱신이 늦어지면 버전 호환이 안될 가능성이 있다.
• serving에 대한 개인적인 문제로, pytorch 말고 tensorflow를 이용하기로 한다.  (3번째 qqwweee/keras-yolov3)
• 아쉬운 점이 있다면 tf2를 쓰고 싶은데, 1이라는 점이다. 

링크 : github.com/qqwweee/keras-yolo3

 

 

python 3.8과 tf 2 로 진행하고 싶다!

• anaconda로 해당 가상환경을 생성한다. (환경 Windows 10, gpu : 1080 ti) 
• keras는 딱히 원하는 버전이 없기 때문에 그냥 설치.

conda create -n yolov3 python=3.8 tensorflow-gpu=2.3 keras

tensorflow 2.3이 잘 적용되었다. 확인하고 생성을 진행한다.

 

linux라면 설치된 라이브러리를 확인하기 위해 grep 을 쓰나 windows에서는 findstr을 쓴다. 

 

제공된 모델파일을 다운받아서 추론을 진행하는 부분. 

• yolov3는 은근히 용량이 크니 빠른 진행을 위해 tiny 모델로 다운로드한다. 
• wget은 linux 명령어라 windows에서 실행되지 않는다. 그래서 설치했던 WSL(2)로 켜서 설치했다. 
  • windows에서 wget을 따로 설치해 진행하려 했으나 SSL 에러가 발생. (추후 해결 과제 스택 +)

 

wget https://pjreddie.com/media/files/yolov3-tiny.weights

모델 인풋으로 줄 이미지를 구해놨다.

원래 여러 마리 동물이 있는 이미지 였으나 test 이미지가 복잡하면 힘들다.

python yolo_video.py --model model/yolov3-tiny.weights --input model_data/test.PNG --image --gpu_num 1 

- PIL 모듈이 없다고 해서 설치 진행 (pip install Pillow) = 8.1.1버전이 설치됐다. 

- mapplotlib이 없다고 해서 설치 진행 (pip install matplotlib) =3.3.4 버전 설치

 

• 코드에 대한 오류가 발생했다. import 된 keras 의 backend 모듈에는 get_session 이라는 항목이 없다고 한다. 예전에 있었던 모듈을 호출한 것이다. 이제부터 버전불일치의 오류가 시작되었다.

버전 바꾸기는 시간 여유가 있을 때 진행하는 것이 좋다.

 

(2)에서 이어진다. 

2021/03/03 - [AI/Github] - Yolov3 (tf1)를 tensorflow 2버전으로 실행해 보자(2)

 

출처 : github.com/qqwweee/keras-yolo3

stackoverflow.com/questions/58255821/how-to-use-k-get-session-in-tensorflow-2-0-or-how-to-migrate-it

www.tensorflow.org/guide/keras/save_and_serialize?hl=ko

www.tensorflow.org/guide/migrate?hl=ko