没有合适的资源?快使用搜索试试~ 我知道了~
YOLOv3-Torch2TRT:将YOLOv3和YOLOv3-tiny(PyTorch版本)转换为TensorRT模型
共58个文件
png:13个
py:11个
jpg:11个
5星 · 超过95%的资源 需积分: 48 14 下载量 142 浏览量
2021-03-13
15:03:40
上传
评论 2
收藏 5.47MB ZIP 举报
温馨提示
YOLOv3-Torch2TRT 介绍 通过torch2trt Python API将YOLOv3和YOLOv3-tiny(PyTorch版本)转换为TensorRT模型。 安装 克隆仓库 git clone https://github.com/DocF/YOLOv3-Torch2TRT.git 下载预先训练的体重 $ cd weights/ $ bash download_weights.sh 要求 需要两个特殊的Python包: 张量 火炬2trt 由于YOLO中的升采样操作,根据torch2trt API的介绍,您需要使用插件安装该版本。 安装参考: : 检查torch2trt API python3 check.py 推理加速 加速技术: FP16 TensorRT 以下是TITAN xp的一些结果: 型号名称 输入尺寸 FP16 整个模式*(FPS) 骨
资源推荐
资源详情
资源评论
收起资源包目录
YOLOv3-Torch2TRT-master.zip (58个子文件)
YOLOv3-Torch2TRT-master
train.py 7KB
utils
utils.py 14KB
datasets.py 5KB
parse_config.py 1KB
__init__.py 0B
__pycache__
parse_config.cpython-36.pyc 1KB
augmentations.cpython-36.pyc 425B
logger.cpython-36.pyc 1KB
datasets.cpython-36.pyc 5KB
utils.cpython-36.pyc 10KB
__init__.cpython-36.pyc 131B
logger.py 674B
augmentations.py 207B
output
messi.png 257KB
room.png 326KB
herd_of_horses.png 275KB
eagle.png 317KB
dog.png 341KB
street.png 309KB
giraffe.png 402KB
person.png 192KB
field.png 316KB
test.py 4KB
detect.py 10KB
check.py 766B
requirements.txt 90B
weights
download_weights.sh 303B
models.py 23KB
config
coco.data 115B
create_custom_model.sh 8KB
custom.data 99B
yolov3.cfg 8KB
yolov3-tiny.cfg 2KB
assets
traffic.png 312KB
messi.png 258KB
dog.png 342KB
giraffe.png 405KB
README.md 2KB
__pycache__
models.cpython-36.pyc 15KB
test.cpython-36.pyc 3KB
data
custom
labels
train.txt 34B
train.txt 29B
images
train.jpg 111KB
valid.txt 29B
classes.names 6B
temptest
field.jpg 111KB
samples
room.jpg 83KB
street.jpg 100KB
messi.jpg 124KB
person.jpg 77KB
field.jpg 111KB
giraffe.jpg 374KB
eagle.jpg 139KB
dog.jpg 160KB
herd_of_horses.jpg 130KB
coco.names 625B
5k.txt 0B
get_coco_dataset.sh 898B
共 58 条
- 1
资源评论
- 汉堡你吃了吗2021-08-19用户下载后在一定时间内未进行评价,系统默认好评。
帝哲
- 粉丝: 34
- 资源: 4669
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功