【文件属性】:
文件名称:EfficientNetv2_TensorRT_int8:EfficientNetv2 TensorRT int8
文件大小:32.03MB
文件格式:ZIP
更新时间:2021-04-07 20:00:42
tensorrt efficientnetv2 Python
EfficientNetv2_TensorRT_int8
EfficientNetv2模型实现来自
环境配置
ubuntu:18.04
cuda:11.0
cudnn:8.0
tensorrt:7.2.16
OpenCV:3.4.2
cuda,cudnn,tensorrt和OpenCV安装包可以从如下链接下载:
链接: 密码: b88e
cuda安装
如果系统有安装驱动,运行如下命令卸载
sudo apt-get purge nvidia*
禁用nouveau,运行如下命令
sudo vim /etc/modprobe.d/blacklist.conf
在末尾添加
blacklist nouveau
然后执行
sudo update-initramfs -u
chmod +x cuda_11.0.2_450.51.05_linux.run
sudo ./cuda_11.0.2_450.5
【文件预览】:
EfficientNetv2_TensorRT_int8-master
----utils.py(2KB)
----effnetv2_tensorrt_int8()
--------models()
--------include()
--------test_imgs()
--------effnetv2_infer.cc(8KB)
--------CMakeLists.txt(1KB)
----test.py(4KB)
----train.py(15KB)
----netron_effnetv2.py(58B)
----effnetv2_tensorrt_int8_tools()
--------util_trt.py(3KB)
--------calibrator.py(2KB)
--------convert_trt_quant.py(3KB)
----.gitignore(2KB)
----README.md(4KB)
----effnetv2.py(5KB)
----export_onnx.py(1KB)