Skip to content

Latest commit

 

History

History

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

README.md

DeepSort (IGIE)

Model Description

DeepSort integrates deep neural networks with traditional tracking methods to achieve robust and accurate tracking of objects in video streams. The algorithm leverages a combination of a deep appearance feature extractor and the Hungarian algorithm for data association.

Supported Environments

GPU IXUCA SDK Release Branch
MR-V100 4.4.0 26.03 release/26.03
MR-V100 4.3.0 25.12 release/25.12

Note: 请切换到与您的 SDK 版本对应的 Release 分支进行测试。请勿直接在 master 分支上运行测试,因为 master 分支可能包含与您的本地 SDK 版本不兼容的最新更改。

切换分支命令示例:git checkout release/26.03

Model Preparation

Prepare Resources

Pretrained model(ckpt.t7): https://drive.google.com/drive/folders/1xhG0kRH1EX5B9_Iz8gQJb7UNnn_riXi6

Dataset: https://zheng-lab.cecs.anu.edu.au/Project/project_reid.html to download the market1501 dataset.

Install Dependencies

pip3 install -r requirements.txt

Model Conversion

python3 export.py --weight ckpt.t7 --output deepsort.onnx

# Use onnxsim optimize onnx model
onnxsim deepsort.onnx deepsort_opt.onnx

Model Inference

export DATASETS_DIR=/Path/to/market1501/

FP16

# Accuracy
bash scripts/infer_deepsort_fp16_accuracy.sh
# Performance
bash scripts/infer_deepsort_fp16_performance.sh

INT8

# Accuracy
bash scripts/infer_deepsort_int8_accuracy.sh
# Performance
bash scripts/infer_deepsort_int8_performance.sh

Model Results

Model BatchSize Precision FPS Acc(%)
DeepSort 32 FP16 17164.67 99.32
DeepSort 32 INT8 20399.12 99.29