파이썬/AI3

• 

  Yolo 객체인식 인식률이 떨어질 경우 확인해야할 것 딸각모드와 인식률이 다르다면? 욜로에서는 기본적으로 detection.py와 같이 딸깍모드로 객체인식 결과를 얻을 수 있습니다.하지만 이 경우 파일이나 동영상으로 결과가 저장되기 때문에 커스텀 알고리즘이 필요하다면 사용하기가 어렵습니다.  좌표 결과를 직접 받아서 실시간으로 처리하려면 이전 포스팅에서 다룬대로 모델 객체를 이용해 직접 이미지를 넣어줘야합니다.  이 때 만약 딸깍테스트의 결과와 직접 model 객체를 사용해서 받은 결과의 인식률, 즉 Confidence 값이 다르다면 의심해봐야할 사항이 하나 있습니다. 테스트 결과 yolo 5, 7버전은 훈련(train) 시 이미지 전처리를 "RGB" 타입으로 합니다. 반대로 yolo 8, 10 버전은 "BRG"형태로 합니다.  따라서 버전에 따라 이미지 .. 파이썬/AI 2024. 6. 20.

• 

  Yolo v5, v8, v10 가중치 파일(pt 파일)로 모델 객체 로딩하는 방법 Yolov5 모델 객체 로딩하기 Yolov5의 모델 객체 로딩은 이전 포스팅에서 다룬 Yolov7과 동일합니다. 아래 코드는 온라인 연결 없이 로컬에서만 로딩하는 방법입니다. Yolov5의 전체 코드를 다운받아서 저장해줘야합니다.Confidence Threshold값은 따로 설정안하면 디폴트가 0.25입니다. 너무 낮으면 오탐이 많아지니 적당한 선에서 설정해주면 됩니다. import torchdevice = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")# 가중치 파일 경로path = r'.\yolov5\save_model\yolov5_medium_head_finetuned_16_30\heads_5m_e30_best.pt' model = t.. 파이썬/AI 2024. 6. 20.

• 

  객체인식 Yolo7(Yolov7) 파인튜닝하는 방법 깃(Git) 클론하기 Yolov7은 ultralytics에서 만든 모델(5,8,10등)과 달리 import 라이브러리가 존재하지 않는 듯합니다. 따라서 먼저 공식 깃허브에서 파일을 다운받아 저장해줍니다. 아래 깃 클론 명령어로 바로 받을 수 있습니다.윈도우 환경의 VSCode 등에서는 git을 pip install로 설치해주고 느낌표를 제외하고 cmd 터미널에서 명령어를 실행해주면 됩니다. # 깃 클론!git clone https://github.com/WongKinYiu/yolov7.git  깃 클론을 할 상황이 아니라면 아래 공식 깃허브 링크에 들어가 초록색 "Code"버튼을 눌러 파일을 다운받은 뒤 압축을 해제해주면 됩니다.  https://github.com/WongKinYiu/yolov7 Git.. 파이썬/AI 2024. 6. 20.