현재 결과이미지가 위와같이 인식은 하지만 이미지에서 객체를 감지한 부분의 클래스 이름들이 너무 겹처버려서 출력부분이 아주 별로임

삽질을 해보았음 (이방법이 맞는지는 모르겠지만 여튼 삽질해봄)


 


이게 위화면에 대한 결과인데 
내가 원하는 결과는 class에있는 값만 뽑아서 cv2.puttext() 함수에 넣고 싶음 


내가 사용한 라이브러리는 파이썬의 cvu-python라는 라이브러리임 

그래서 여기서 지원해주는 draw라는 함수를 들어가봄



근데 봐도 뇌정지가 와서 모르겠음





근데 해당함수가 있는 파일이름이 이거더라고 ? 

그래서 패키지 불러오는 import 문에서 
"prediction에 뭔가 있겠다!"라는 나만의 뇌피셜로 라이브러리를 들어가봄



그래서 들어가보니 이런 파일이 또 있었음

그래서 draw_bbox 클릭해서 들어가봄



그랬더니 draw.py 파일이 나와서 

putext에 들어가는 정보가 title 이라는 변수로 잡혀 있더라고 


예를 프린트 해보니까


이런 정보가 들어가더라고



 

여기서 내가 원하는건 "Jordan1" 출력하고 싶으니까 

이를 어떻게 짜를까 고민해봄 



일단 내가 만든 모델의 예측범주는 이딴식으로 생김.

['jordan11','jordan1','jordan4','jordan2','jordan3','jordan5','jordan6']

보면 jordan+"숫자" 만 있는 형태니까 

이부분을 정규표현식으로 패턴화 하면 될것같다! 라는 결론을 내려 

이를 사용하기로 하였음 


코드는 아래와 같이 추가만해줌
pattern은 jordan뒤에 오는 넘버링의 자릿수가 1자리 또는 2자리니까 이렇게 패턴을 잡고 

scale값도 조정해봄



결론은 

CVU 패키지의 utils 파일에 있는 draw.py 부분의 title 부분만 건드려서 바꿈


다시 돌려보니까 아주 깔끔하게 나옴 

테스트 코드도 끝났으니까 라즈베리 파이 서버에 올려야겠음



필력 ㅆㅎㅌㅊ라 ㅈㅅ