pytorch中的坑
Zio_wh
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
Anaconda修改镜像源安装PyTorch
Conda 修改镜像源查看目前已有的channel: conda config --show channels添加新的channel:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://...原创 2019-12-10 20:27:02 · 4308 阅读 · 0 评论 -
pytorch中使用transforms.ToTensor()处理opencv获取的灰度图像问题
pytorch中使用transforms.ToTensor()处理opencv获取的灰度图像问题众所周知,opencv读取的图片的格式为BGR即(height,width,channel),而PIL读取的图片恰恰相反,为(channel,height,width),matplotlib中也是如此。在此插一句:如果要使用matplotlib显示opencv读取的图片,需要将opencv读取的图片...原创 2019-03-13 16:09:00 · 9152 阅读 · 0 评论 -
PyTorch读取模型权重:IncompatibleKeys(missing_keys=[], unexpected_keys=[])
IncompatibleKeys(missing_keys=[], unexpected_keys=[])本人在服务器上训练好一个模型后,进行测试时,读入权重文件.pkl(/.t7)时,出现该错误,通过查询源文件torch/nn/modules/module.py中的load_state_dict(),只需将第778行的return _IncompatibleKeys(missing_keys...原创 2019-08-13 11:19:34 · 9442 阅读 · 0 评论
分享