This project aims at performing image classifications (Monkey dataset and Cars dataset) using transfer learning in deep neural networks along with TensorRT inference acceleration.

1. Pytorch (GPU Version)
2. TensorRT 5.x.x

Using Google Colab to run jupyter notebook files is highly recommended.

1. training:
   Open ResNet_152_Testing_on_Cars_Dataset.ipynb and run all.
2. testing:
   Open ResNet_152_Train_on_Cars_Dataset.ipynb and run all.

1. Save model to xx.onnx file
2. python cars_152_build_engine.py to create tensorRT engine.
3. python cars_152_run.py to run the inferencce acceleration.

andrewhuang51
arikhor
eebowen
hjyu2019
yeqinghuang516