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

AI/Github

아인샴 2021. 3. 3. 03:23

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

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

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

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

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

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

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

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)에서 이어진다.