snehitvaddi / yolov3-object-tracking

Object tracking with YOLOv3, YOLOv3-tiny, Deep Sort and Tensorflow. This repository implements YOLOv3 and Deep SORT in order to perfrom real-time object tracking.

Object Tracking using YOLOv3, Deepsort and Tensorflow

This repository implements YOLOv3 and Deep SORT in order to perfrom real-time object tracking. Yolov3 is an algorithm that uses deep convolutional neural networks to perform object detection. We can feed these object detections into Deep SORT (Simple Online and Realtime Tracking with a Deep Association Metric) in order for a real-time object tracker to be created.

🛠 Video Example

📋 Example

Detailed tutorial by @The AI Guy on Object TrackingYoutube Tutorial.

🧠 Original Repo	💡 Colab Notebook
Github	Colab

🏃‍♂️ Getting started

📥 Pip

# TensorFlow CPU
pip install -r requirements.txt

# TensorFlow GPU
pip install -r requirements-gpu.txt

📥 Conda (Recommended)

# Tensorflow CPU
conda env create -f conda-cpu.yml
conda activate tracker-cpu

# Tensorflow GPU
conda env create -f conda-gpu.yml
conda activate tracker-gpu

🛠 Nvidia Driver (For GPU, if you haven't set it up already)

# Ubuntu 18.04
sudo apt-add-repository -r ppa:graphics-drivers/ppa
sudo apt install nvidia-driver-430
# Windows/Other
https://www.nvidia.com/Download/index.aspx

📥 Downloading official pretrained weights

For Linux: You can download official yolov3 weights pretrained on COCO dataset.

# yolov3
wget https://pjreddie.com/media/files/yolov3.weights -O weights/yolov3.weights

# yolov3-tiny
wget https://pjreddie.com/media/files/yolov3-tiny.weights -O weights/yolov3-tiny.weights

If you are on Windows, you can directly download the YOLOv3 weights from 👉 here

Using Custom trained weights

Learn How To Train Custom YOLOV3 Weights Here: https://www.youtube.com/watch?v=zJDUhGL26iU

Add your custom weights file to weights folder and your custom .names file into data/labels folder.

Saving your yolov3 weights as a TensorFlow model.

Load the weights using load_weights.py script. This will convert the yolov3 weights into TensorFlow .tf model files!

# yolov3
python load_weights.py

# yolov3-tiny
python load_weights.py --weights ./weights/yolov3-tiny.weights --output ./weights/yolov3-tiny.tf --tiny

# yolov3-custom (add --tiny flag if your custom weights were trained for tiny model)
python load_weights.py --weights ./weights/<YOUR CUSTOM WEIGHTS FILE> --output ./weights/yolov3-custom.tf --num_classes <# CLASSES>

After executing one of the above lines, you should see proper .tf files in your weights folder. You are now ready to run object tracker.

Running the Object Tracker

Now you can run the object tracker for whichever model you have created, pretrained, tiny, or custom.

# yolov3 on video
`python object_tracker.py --video ./data/video/test.mp4 --output ./data/video/results.avi`

#yolov3 on webcam 
`python object_tracker.py --video 0 --output ./data/video/results.avi` (May not properly in Colab)

#yolov3-tiny 
`python object_tracker.py --video ./data/video/test.mp4 --output ./data/video/results.avi --weights ./weights/yolov3-tiny.tf --tiny`

#yolov3-custom (add --tiny flag if your custom weights were trained for tiny model)
python object_tracker.py --video ./data/video/test.mp4 --output ./data/video/results.avi --weights ./weights/yolov3-custom.tf --num_classes <# CLASSES> --classes ./data/labels/<YOUR CUSTOM .names FILE>

The output flag saves your object tracker results as an avi file for you to watch back. It is not necessary to have the flag if you don't want to save the resulting video.

There is a test video uploaded in the data/video folder called test.mp4. If you have followed all the steps properly then you should see the output as below by running first command.

python object_tracker.py --video ./data/video/test.mp4 --output ./data/video/results.avi

---

📧 Acknowledgments

• Yolov3 TensorFlow Amazing Implementation
• Deep SORT Repository
• Yolo v3 official paper

yolov3-object-tracking's Issues

Error while executing object_tracker.py in COLAB

W0315 07:24:22.164951 140703536494464 object_tracker.py:83] Empty Frame
W0315 07:24:22.265901 140703536494464 object_tracker.py:83] Empty Frame
W0315 07:24:22.366825 140703536494464 object_tracker.py:83] Empty Frame
Traceback (most recent call last):
File "object_tracker.py", line 163, in
app.run(main)
File "/usr/local/lib/python3.7/dist-packages/absl/app.py", line 312, in run
_run_main(main, args)
File "/usr/local/lib/python3.7/dist-packages/absl/app.py", line 258, in _run_main
sys.exit(main(argv))
File "object_tracker.py", line 155, in main
if FLAGS.ouput:
File "/usr/local/lib/python3.7/dist-packages/absl/flags/_flagvalues.py", line 481, in getattr
raise AttributeError(name)
AttributeError: ouput

No result

Thank！

Issue running in Colab

Hi man, when running "object_tracker.py" in Colab, it gives me this error: "cannot connect to X server". It is in fact in your Colab. The thing is that I cannot convert the video to mp4 because the "traffic-results2.avi" is empty, damaged. If I download it, it would't let me open it.

Any idea why? Thanks!