Skip to content

yolov3_tiny(add SE model)(pytorch 1cls for car),deep_sort(pytorch),mx150 GPU, 14 avg_fps

License

Notifications You must be signed in to change notification settings

ggyybb/yolov3-tiny-deepsort-pytorch-cardetection

Repository files navigation

Deep Sort and yolov3_tiny for car detection with PyTorch

nearly realtime:avg_fps=14 for mx150(2G GPU)

Latest Update(2020-3-9)

Weights files and demo.mp4 at here:(also i put this video to bilibili,after passing the audit,i will put 链接) [BaiduDisk] 链接:https://pan.baidu.com/s/1czw0dz56ZjfgrxjBFaUSdg 提取码:odbu Yolov3_tiny_1cls_se detection results at here: 链接:https://www.bilibili.com/video/av94365366/

Dependencies

  • python 3
  • numpy
  • scipy
  • opencv-python
  • sklearn
  • torch ==1.1.0
  • torchvision ==0.3.0
  • pillow
  • vizer
  • edict

Quick Start

  1. Check all dependencies installed

  2. Download best.pt(yolo weights) parameters

look up
  1. Download deepsort parameters ckpt.t7
look up and put here:
cd deep_sort/deep/checkpoint

Notice: If compiling failed, the simplist way is to **Upgrade your pytorch >= 1.1 and torchvision >= 0.3" and you can avoid the troublesome compiling problems which are most likely caused by either gcc version too low or libraries missing.

  1. Run my_deep_sort.py
my is windows
i do not know linux or ubuntu
my programming ability is very weak
comments are not very good
forgive me

Notice: Maybe some mistakes, you can discuss in issues, help me to modify. It is just a small work, you can improve based on my code.

References

About

yolov3_tiny(add SE model)(pytorch 1cls for car),deep_sort(pytorch),mx150 GPU, 14 avg_fps

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages