python
关注
分享
扫一扫
工长山
求推荐深度学习工程师，我的简历：https://raw.githubusercontent.com/basaltzhang/cv_zhangyifeng/master/cv_%E5%BC%A0%E5%B1%B9%E5%B3%B0.pdf
展开
-
复杂模型的Pytorch到Tensorrt转换以及Plugin编写
文章目录背景准备工作简单案例Pytorch->ONNXONNX->TensorRT复杂案例编译自定义算子并运行demo转换为onnx准备步骤转换代码异常解析自定义算子转换为Tensorrt转换脚本异常解析之topK修改pytorch源码修改tensorrt源码重新转换异常解析之NonZero自定义算子NMS加入自定义torch层修改torch调用修改TensorRT源码自定义算子RoiAlign（略）运行tensorrt模型运行脚本异常解析总结背景在实际工作当中，训练的模型到实际使用还需要有原创 2020-12-20 22:24:25 · 9428 阅读 · 26 评论 -
Convnet生成batches.meta等数据文件
最简单的方法是找个能调试的python IDE，读取各个文件之后看里面的内容，这里有个我写的方法，仔细阅读之后就可以使用了。事先安装PIL包。有错写评论即可。第93,94行是建立各个data_batch_1等等最后一行建立batches.meta文件import osimport cPickleimport pickleimport numpy as npfrom nu原创 2013-11-25 18:14:31 · 4480 阅读 · 38 评论 -
Convnet中由输入图片测试数据并输出结果
原作者Alex只是随机的选取数据中的6幅图预测最大的预测分类原创 2014-04-16 21:26:04 · 2836 阅读 · 3 评论 -
Alexnet在Windows下的使用
有个外国人也实现了这个东西，逻辑比较清楚，看不懂我的文章的话，移步http://personal.ie.cuhk.edu.hk/~xy012/others/install_cudaconvnet.html写在前面前一阵子百度有个ICOME比赛，任务一火眼金睛主要内容就是对图像有人没人进行分类，师兄说CNN不错，于是用了Alex Krizhevsky的pro...原创 2013-10-16 20:19:19 · 19542 阅读 · 87 评论