opencv & yolov5 와 pyqt5 를 이용한 데이터 전, 후 처리 작업관련 질문입니다!

Question

안녕하세요!예전 개발 취업이 왜이리 안될까 하며 주저리주저리 글을 올렸던 새내기입니다.많은 조언 주신덕분에 취업을 하게되고 프로젝트를 하게 되었습니다그런데 처음 해보는 부분이라 그런지 서치 능력이 아직 부족해서 그런지 해답을 못찾고 있어 선배님들께 도움 청하고자 이렇게 글 작성해봅니다먼저 제가 하는 일은 이미지 프로세싱으로 Yolov5 검사 모델에 Frame 데이터 입력 전 처리와 검사 모델 입력 후 출력되는 데이터 후 처리 (bbox, label) 입니다.여기서 Pyqt5 를 이용해 GUI로 화면 출력하는것 까지는 되었는데 문제는 영상에서 객체 인식 부분이 제대로 인식을 못하고 있다게 문제입니다;;;어디가 문제인지 확인하기 위해 여기 저기 프린트도 찍어 보았으나 어떤 문제인지 감이 안와 도와주시면 감사하겠습니다!우선 작업한 사안입니다.입력 이미지 사이즈 resize정규화 (0,255) -> (0,1)그후 이미지 텐서 4차원 으로 조절검사 모델에 입력출력된 데이터를 이용하여 처음 입력 이미지에 bbox 와 label 그려주기pyqt5 화면에 출력을 위해 frame BGR 2 RGB로 전환위와 같이 작업한 후 실행시 화면은 출력이 되나 물건을 인식하는 부분에서 엄한 곳을 찾아서 바운딩 박스를 그려지고 있습니다.그래서 해결해보기 위해 입력 이미지 사이즈 변경 -> 실패 (pt에서 정해 놓은 사이즈가 있다고 해서 원래 설정으로 돌림)출력된 데이터를 이용하여 처음 입력 이미지가 아닌 원본 이미지 (cv2.VideoCapture() 에서 read() 를 통해 받아온 데이터)에 bbox와 label 그려보기 -> 화면이 잘리면서 GUI에 출력되고 물건 인식도 해결 되지 않음텐서 차원 부분 다른 차원 으로 바꿔보기 -> 기본적으로 Yolov5의 경우 4차원 텐서를 이용한다는걸 깨달으며 에러가 발생하고 다시 원복귀위의 1번과 2번을 동시에 실험 -> 실패와 화면 잘림, 물건 인식 해결 안됨위와 같이 해결해 보기위해 해보았으나 해결 되지가 않습니다... 선배님들께서는 이런경우가 있었다면 어는 부분을 수정하여 고치셨는지 어떻게 고치셨는지 이런경우 어떤식으로 검색을 하는지 소스를 주신다면 정말 감사하겠습니다.그럼 마지막으로 항상 좋은 하루되시길 바라며 물러나 보겠습니다.&#43;코드는 보안상의 이유로 바깥으로 가지고 오지 못하여 올리지 못하였습니다! 가독성이 떨어지는점 죄송합니다!

hmmm · Answer

read로 읽어 온 frame -> 전처리 -> Yolo -> result 이 시퀀스로 하신게 맞으신가요? 맞다면 fps 조절을 해보세요. 카메라가 읽어오는 frame 속도를 이미지 처리 하는 과정이 따라가지 못해 딜레이가 생겼을 수 있습니다

opencv & yolov5 와 pyqt5 를 이용한 데이터 전, 후 처리 작업관련 질문입니다!

1개의 답변

Q&A 목록