GitHub - lshiwjx/Short-Video-Classification: fast short video classification with mxnet, tensorrt

lshiwjx / Short-Video-Classification Public

Notifications You must be signed in to change notification settings
Fork 2
Star 2

fast short video classification with mxnet, tensorrt

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
aic_tensorrt_acap_2s		aic_tensorrt_acap_2s
model		model
.gitignore		.gitignore
README		README
int8calibration.py		int8calibration.py
mxnet_to_onnx.py		mxnet_to_onnx.py
mytrain_rcap.py		mytrain_rcap.py
myval_multilabel.py		myval_multilabel.py
onnx_to_trtengine.py		onnx_to_trtengine.py
plot_curve.py		plot_curve.py

Repository files navigation

# Requirements

MXNet
TensoRT=5.0
ONNX=

# Usage

mytrain_rcap.py: train mxnet model
myval_multilabel.py: validate mxnet model
plot_curve.py: plot acc curve
model save the model files

mxnet_to_onnx.py: convert mxnet to onnx
onnx_to_trtengine.py: convert onnx to tensorRT engine
int8calibration.py: calibrate onnx model and transform it to in8 tensorRT engine

aic_tensorrt_acap_2s is a example of the docker image construction process