PyTorch_Projects

Pretrained model repository for Heterogeneous Architecture Models on Pytorchwith CPU/CUDA

NOTE : Jetson TX2 Development kit (Nvidia Tegra X2 GPU) with Jetpack 3.3 is used for these results

Source code execution : python Model.py --gpu 0/1 --file /Path/to/Image

For more paremeters type: python Model.py --help

Example: cd AlexNet python AlexNet.py --gpu 1 --file cat.jpg

10/01/2019 : Currently just partitioning between CPU/GPU per layer on LeNet5 and Fully CPU or fully GPU on AlexNet and ResNet(152)
23/01/2019 : Able to get video/images from terminal using CSI Camera. However, Gstreamer is not supported by current OpenCV L4T version on Jetson TX2. OpenCV must be recompiled supporting Gstreamer
13/10/2019 : Added GPU power consumption graphics

MIT License

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
AlexNet		AlexNet
LeNet5		LeNet5
ResNet152		ResNet152
README.md		README.md