
pytorch
ning_ww
这个作者很懒,什么都没留下…
展开
-
pytorch3d安装 Output in format: Requested package -> Available versionsThe following specifications
一、问题描述在pytorch3d的安装过程中,反复尝试不同版本cuda下的pytorch3d的安装,均显示环境冲突无法安装,而且报出相同的错误,如下:在对比github issue中他人的环境信息后发现gcc版本不对应,修改后即安装成功。UnsatisfiableError: The following specifications were found to be incompatible with each other:Output in format: Requested package ->原创 2022-03-21 10:15:00 · 4605 阅读 · 8 评论 -
图像ROI区域如何映射到feature map—Mask R-CNN
问题描述某些情况下需要将卷积/池化层后的features map的某些patch映射回原图像,就需要features map怎么变过来的怎么回去。#read imageself.object_detector = torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained = True, trainable_backbone_layers = 0)images_nested = torchvision.models.detection.原创 2022-01-27 13:07:32 · 3398 阅读 · 0 评论 -
torchvision自动下载预训练模型后的保存路径—urllib.error.URLError: <urlopen error [Errno -2] Name or service not know
问题描述集群中计算节点无法联网,或者在某些网络不太好的情况下使用torchvision中预训练模型加载失败,这时候就需要我们离线下载模型pth文件后再迁移到服务器中正确的路径。解决方法这里以maskrcnn预训练模型为例,其他同理。1.获取预训练模型网址并下载:Anacondas/anaconda3/envs/envsname/lib/python3.7/site-packages/torchvision/models/detection/mask_rcnn.py代码中搜索关键字model_u原创 2022-01-13 21:47:34 · 1324 阅读 · 0 评论 -
TypeError: Object of type Tensor is not JSON serializable
测试模型时,保存检测结果时出现这个错误:TypeError: Object of type Tensor is not JSON serializable网上提供了一种解决方案是这样的:#json.dump(detections.numpy, file_obj)json.dump(detections.numpy(), file_obj)引发的又是另一个错误:AttributeError: 'list' object has no attribute'numpy'Emmm…最终决定还是原创 2021-12-13 10:53:32 · 4790 阅读 · 0 评论 -
OSError(28, ‘No space left on device‘)
最近因为要适配代码的环境,pytorch包在conda环境中下载的有点多,数次下载后出现这个问题解决方法打开anaconda3/pkgs文件夹,会发现一堆安装包,只需要删掉几个不用的就可以了即使在环境中uninstall了以后,安装包还会缓存在这里面的,所以这个还是必要清理一下的!...原创 2021-11-17 17:19:02 · 1465 阅读 · 0 评论 -
RuntimeError: Non RGB images are not supported.
版本问题torchvision版本过低,io.read_image不支持灰度图原创 2021-11-17 12:37:26 · 552 阅读 · 0 评论 -
AttributeError: module ‘torchvision.io‘ has no attribute ‘read_image‘
版本问题再现,最近在搬运gayhub上的一个代码,但是没给版本信息,导致报了很多错,正在不断尝试ing。。。You could update your torchvision library, as this method was introduced in 0.8.0 so I guess you are using an older version....原创 2021-11-17 12:24:25 · 2318 阅读 · 0 评论 -
python [m::n],[...,i]用法
在搬运别人代码时遇到了两个没见过的索引方式,自己试验之后,大概搞懂了1、[m::n]a = [1,2,3,4,5,6,7,8]>>>b=a[3::1]>[4,5,6,7,8]>>>b=a[3::2]>[4,6,8]这里我们就可以推断出,这是省略索引终值的一种列表索引方式,m为起点,n为步长,省略的终值就默认为列表末尾。2、[…,j]这种用法是在JitterBoxes的代码中第一次见到的,结合代码中的注释和个人测试...原创 2021-11-17 12:14:40 · 1716 阅读 · 0 评论 -
ImportError: cannot import name ‘tensorboard‘ from ‘torch.utils‘ (unknown location)
记录一个困扰我很久的问题。。。又是一个在网上找不到解决方案的问题。。。原创 2021-11-15 14:30:28 · 5336 阅读 · 1 评论 -
F.nll_loss与F.cross_entropy(loss为nan时如何调整)
在用vgg16原创 2021-02-01 18:43:05 · 3552 阅读 · 1 评论