furiosa-ai / smart-cctv-demo Goto Github PK

sudo apt install cmake
sudo apt install \
  furiosa-libhal-warboy=0.9.0-2+nightly-220810 \
  libonnxruntime=1.11.1-2 \
  furiosa-libnux=0.8.0-2+nightly-220810 \
  furiosa-libcompiler=0.8.0-2+nightly-220803
conda env create -n "cctv" python=3.8
conda activate cctv
conda install cython
conda env update --file environment.yml
bash build.sh

smart-cctv-demo$ tree -L 2 data weights
data
├── CrowdHuman
│   └── Images
├── PRW-v16.04.20
│   ├── ID_test.mat
│   ├── ID_train.mat
│   ├── annotations
│   ├── frame_test.mat
│   ├── frame_train.mat
│   ├── frames
│   ├── generate_query.m
│   ├── query_box
│   ├── query_info.txt
│   └── readme.txt
├── WIDER_val
│   └── images
├── market1501
│   ├── Market-1501-v15.09.15
│   └── Market-1501-v15.09.15.zip
├── ms1m-retinaface-t1
│   ├── agedb_30.bin
│   ├── cfp_fp.bin
│   ├── lfw.bin
│   ├── property
│   ├── train.idx
│   ├── train.lst
│   └── train.rec
├── tom_cruise_profile1.jpeg
└── tom_cruise_test.mp4
weights
├── ms1mv3_r50_leakyrelu_1.pt
├── resnet50_market_xent.pth.tar
├── yolov5_face_relu.pt
└── yolov5m_warboy.pt

# Optionally clear cache
rm *.onnx  # f32 model cache
rm *.dfg  # i8 model cache
rm -rf galleries/*  # gallery cache
rm -rf results/*  # delete previous results

# Face recognition (results will be written to "results/face")
cd insightface
python recognition/arcface_torch/query_video.py \
  --gallery ../data/tom_cruise_test.mp4 \
  --query ../data/tom_cruise_profile1.jpeg \
  --device furiosa

# ReId (results will be written to "results/reid")
cd torchreid
python query_video.py \
  --gallery ../data/PRW-v16.04.20/frames \
  --query ../data/PRW-v16.04.20/query_box/479_c1s3_016471.jpg \
  --device furiosa