szemenyeim / DeepLearningLab

Control Engineering and Image Processing Lab - Object Tracking

CIFAR 10 training and object tracking demo in PyTorch

Control Engineering and Image Processing Lab - Object Tracking

Requirements

• Python 3.6
• Pytorch 4.x
• OpenCV 3.x
• Optional: Tensorflow (for TensorBoard logging)

Run

For the CIFAR 10 training run python train.py

For the tracking use python track.py

References

Guided Backprop code based on https://github.com/utkuozbulak/pytorch-cnn-visualizations

DenseNet implementation by https://github.com/kuangliu/pytorch-cifar