Safemotion Lib
Loading...
Searching...
No Matches
smreid

  • fast-reid을 기반으로 필요 기능 구현

데모 실행

데모 준시사항

  1. fast-reid 기반으로 학습한 모델 필요
    • 모델 config 파일 준비 (.yaml)
    • 모델의 파라미터 준비 (.pth)
  2. 데모에 사용할 크롭 이미지 준비(사람의 영역을 자른 이미지)

실행 방법

  1. 데모스크립트('reid_demo.py') 수정
    • 동작할 디비이스 설정
    • 모델의 cfg 경로 설정
    • 모델의 파라미터 경로 설정
    • 데모에 사용할 이미지 폴더 경로 설정
  2. 데모스크립트 실행
    python reid_demo.py

TODO

  1. TensorRT 변환 및 변환 모델 사용 기능 구현
  2. 자체 모델 개발