Mxnet Image Classiciation TensorRT

Hello, guy. I have a small project with Image Classification convert to TensorRT Engine and runtime. My project using mxnet deep learning framework pre-train ImageNet convert to TensorRT Engine. Device I test on is Jetson Nano include Jatpack4.5.1. In my repo also have a Fine tuning with pre-train ImageNet Demo and Deploy to TensorRT Engine.

Code : jetson_nano_mxnet_tensorrt

Example

Inference with Raspberrypi Camera V2 resnet18_v1 FP16 Engine
gif_imagenet

Inference with Image resnet18_v1 FP16 Engine

Fine Tuning with Custom dataset Inference with CSI
board

Fine Tuning with Custom dataset Inference with Image

1 Like