pytorch
街道口扛把子
武汉大学炼丹师
展开
-
pytorch DCNv2使用方式
主要注意offset的获得方式:from torchvision.ops import DeformConv2dclass DConv(nn.Module): def __init__(self, inplanes, planes, kernel_size=3, stride=1, padding=1, bias=False): super(DConv, self).__init__() self.conv1 = nn.Conv2d(inplanes, 2 *原创 2021-12-18 22:06:00 · 3679 阅读 · 3 评论 -
onnxruntime模型部署流程
一、将训练好的模型转换格式为ONNX格式例如pytorch模型转换:def torch2onnx(model, save_path): """ :param model: :param save_path: XXX/XXX.onnx :return: """ model.eval() data = torch.rand(1, 3, 224, 224) input_names = ["input"] output_names =原创 2020-10-03 10:22:45 · 5351 阅读 · 3 评论 -
Qt+libtorch+PSENet
https://github.com/Whu-wxy/PSENet-libtorch能正常运行,但是结果不正确,还存在一些问题。原创 2020-02-09 21:56:27 · 401 阅读 · 0 评论 -
Qt+libtorch+YOLO
处理单张图:https://github.com/Whu-wxy/libtorch-yolov3/tree/master/libtorch-yolov3处理视频:https://github.com/Whu-wxy/libtorch-yolov3/tree/master/VideoVLC原创 2020-02-09 21:55:24 · 1176 阅读 · 0 评论 -
Adam,Ranger,RAdam,Lookahead对比
mgrankin/over9000https://github.com/mgrankin/over9000原创 2019-10-05 10:35:47 · 919 阅读 · 0 评论 -
PSENet.pytorch使用中遇到的问题以及解决,优化
源码总体写的还是非常棒的,反正我是不能完全地写出来。学习一下。开源地址:https://github.com/WenmuZhou/PSENet.pytorch在这几天使用的过程中,发现了一些bug并进行了修改,对一些细节进行了优化,在这里记录下来。issue地址:https://github.com/WenmuZhou/PSENet.pytorch/issues/39eval失败。解决:s...原创 2019-10-04 15:02:01 · 1697 阅读 · 16 评论 -
BERT的vocabulary字典
在pytorch-pretrained-BERT/pytorch_pretrained_bert/tokenization.py文件中可以看到BERT使用的vocabulary链接,但是不用特殊的上网方式打不开。PRETRAINED_VOCAB_ARCHIVE_MAP = { 'bert-base-uncased': "https://s3.amazonaws.com/models.hu...原创 2019-03-17 09:40:17 · 9673 阅读 · 4 评论 -
Pytorch快速离线安装
    前一阵配置Pytorch的虚拟环境,即使加了清华镜像也好像跳到了默认路径,装载龟速,更可恶的是中间会挂掉,无奈之下,开始研究如何快速安装,终于找到了(泪奔)~~~~~ 首先,到https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/中下载所需的.tar转载 2019-03-08 10:17:55 · 3568 阅读 · 0 评论 -
NLP的强大工具——AllenNLP
学习NLP,推荐一下AllenNLP这个工具。从学习pytorch到学习处理NLP问题,中间有很长的路要走,其中有很多坑。NLP数据的预处理要比CV的麻烦很多,比如去除停用词,建立词典,加载各种预训练词向量,Sentence -> Word ID -> Word Embedding的过程(文本预处理方法小记)。用到RNN时,还要经过pad,pack,pad的过程。还有数据集的读取,t...原创 2019-01-17 16:00:49 · 13266 阅读 · 2 评论 -
推荐Github中的一些machine reading模型
这些都是pytorch的实现。第一次尝试运行的是DrQA,计划接下来几天试一下SAN,没问题的话假期读一下SAN的代码。还可以看一下后面的QANet和BIDAF,之前大概浏览了一下,代码量少,对新手应该友好一些。SAN:https://github.com/kevinduh/san_mrcmnemonic:https://github.com/HKUST-KnowComp/Mnemonic...原创 2018-12-17 11:47:55 · 673 阅读 · 0 评论 -
尝试使用CMake构建Qt+Pytorch简单应用
CMake+libtorch+Qt毕业设计打算做一个小软件,实现一个简单的Machine Reading。前几天pytorch发布了1.0稳定版,据说在部署上更方便,于是尝试一下。首先准备anaconda+pycharm+Qt+pytorch1.0+libtorch。首先将pytorch官方的示例在ubuntu中做了一遍,完成之后,需要的是把Qt做的界面加入到里面,走了一些弯路,在windo...原创 2018-12-12 18:41:50 · 3531 阅读 · 4 评论