[flask] yolov5 와 flask rest api 서버 연동하기

졸업작품 프로젝트를 진행하면서 yolov5 를 이용한 객체 탐지 결과를 띄워주는 서버가 필요해서 flask 를 이용해 구현해보았다.

 

서버를 띄우기 전에 여러가지를 설치해야한다.

 

https://github.com/robmarkcole/yolov5-flask/blob/master/requirements.txt

GitHub - robmarkcole/yolov5-flask: Official implementation at https://github.com/ultralytics/yolov5/tree/master/utils/flask_rest

 

위의 깃허브 링크에서 requirments.txt를 다운 받아서  설치해주었다.

 

설치가 끝났으면, yolov5 폴더에 들어있는 메뉴얼 파일을 수정해주어야한다.

 

여기서 하나 오류가 발생했는데 시간을 꽤나 잡아먹었다.

근데 굉장히 해결하기 쉬운 오류였다 ...

앞으론 설치할 때 경고문도 꼼꼼히 봐야겠다.

 

https://cb036133.tistory.com/69

[flask] The script flask is installed in 'PATH' which is not on PATH. 경고가 뜰 때

 

​기존 코드는 그대로 사용해도 되는데 아래의 코드를 수정해주어야한다.

모델은 custom모델을 쓰고 모델이 들어있는 경로를 입력해주었다.

 

설정이 끝났으면 서버를 실행해준다.

 

다른 터미널로 접속해 명령어를 입력한다.

curl -X POST -F image=@pic1001.jpg 'http://localhost:5000/v1/object-detection/yolov5'

우리 팀의 모델 이름은 yolov5이므로 마지막에 yolov5를 넣어주었다.

​이렇게 명령어를 입력하고 실행하면 .. !

좀 정신없긴 하지만 json형식으로 xmin , ymin , xmax, ymax, confidence , class, name 같은 정보들이 잘 출력되었다.

 

이제 이것들을 이용해서 메뉴얼에서 제공하는 데이터가 아닌 프로젝트에 필요한 데이터를 받는 법을 연구해봐야될 것 같다