Yolov5_DeepSort_Pytorch

Yolov5_DeepSort_Pytorch

제품 정보

즐겨찾기

프로그래밍 언어

공개 채팅

지원 계획

현재 사용할 수 있는 OSS 플랜이 없습니다.

저장소의 제공자 또는 기여자인 경우 OSS 플랜 추가를 시작할 수 있습니다.

OSS 플랜 추가
여기에서 자세히 알아보세요

이 오픈소스에 대한 플랜을 찾고 있다면 저희에게 문의해 주세요.

전문 공급자와 연락하실 수 있도록 도와드리겠습니다.

제품 세부 정보

Real-time multi-object tracker using YOLO v5 and deep sort

Easy training program of Yolov5 Deep Sort with PyTorch

Prerequisites

  • Already installed Docker

How to use

  • create environment (common step)

    • execute following script in command line
      
      # clone this repository
      git clone --recurse-submodules https://github.com/tmfi-analytics/Yolov5_DeepSort_Pytorch.git

    build Docker container using Dockerfile

    cd Yolov5_DeepSort_Pytorch docker build -t [YOUR CONTAINER NAME] .

3 ways to use

  • Just execute human tracking demo

    • edit config.sh as following

      # 訓練済みのモデルで物体追跡をする場合はここを1に設定
      JUST_PREDICTION=1
      
      # YOLOV5の訓練に自前のデータセットを使用する場合はここを1に設定
      USE_CUSTOM_DATASET=0
    • execute docker run command
      cd Yolov5_DeepSort_Pytorch
      docker run -it --env-file=config.sh -v `pwd`/data:/app/data --shm-size=2048m [YOUR CONTAINER NAME]
    • watch the output video
      ls data/result
  • train the human detection model and track using prepared dataset

  • train the human detection model and track using custom dataset

    • put train and val dataset as following
        スクリーンショット 2021-12-14 23 36 52
      put image files in images directory and label files in labels directory. the pair of image file and label file must be same file name
      please refer the following link about label format
      https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
    • put mp4 file for test as data/test/test.mp4
    • edit config.sh as following

      # 訓練済みのモデルで物体追跡をする場合はここを1に設定
      JUST_PREDICTION=0
      
      # YOLOV5の訓練に自前のデータセットを使用する場合はここを1に設定
      USE_CUSTOM_DATASET=1
    • execute docker run command
      cd Yolov5_DeepSort_Pytorch
      docker run -it --env-file=config.sh -v `pwd`/data:/app/data --shm-size=2048m [YOUR CONTAINER NAME]
    • watch the output video
      ls data/result

문의하기

  • 비즈니스 개발자를 위한 최신 정보를 원하십니까? 소스 코드 프로젝트에 대한 PieceX 커뮤니티의 요구사항을 알아보세요. PieceX의 최신 무료 커뮤니티 코드를 빠르게 알려드립니다.
PieceX Logo