Pytorch
pytorch框架
wait a minutes
创作不易,可以订阅博主哦。
展开
-
【多目标跟踪】ByteTrack复现(有yolov5例子)
1.环境ubuntu16.04cuda10.1cudnn7python3.6 Cythonmatplotlib>=3.2.2numpy>=1.18.5opencv-python>=4.1.2PillowPyYAML>=5.3scipy>=1.4.1tensorboard>=2.2torch>=1.7.0torchvision>=0.8.1tqdm>=4.41.0seaborn>=0.11.0easydic原创 2021-11-25 21:32:52 · 9051 阅读 · 30 评论 -
【caffe】关于gempool层
gempoolP torch转caffe原创 2021-10-24 09:00:00 · 1288 阅读 · 0 评论 -
【pytorch】pytorch读模型打印参数
import torchfrom collections import OrderedDictfrom your_model import Net# your net architecturenet = Net()model_path = "your_model_path"# load model parametersstate_dict = torch.load(model_path, map_location="cpu")# define a new dictnew_state_.原创 2021-10-16 11:06:52 · 956 阅读 · 0 评论 -
【Pytorch】自定义DataLoader相关分析
1.三步法写自定义Torch的DataLoader - 知乎2.pytorch Dataset, DataLoader产生自定义的训练数据_pan_jinquan的博客-CSDN博客3.GitHub - JDAI-CV/fast-reid: SOTA Re-identification Methods and Toolbox4.https://github.com/chenyuntc/simple-faster-rcnn-pytorch第一个链接,会给出一个大体的框架思路,第二个链接,是一个原创 2021-09-14 22:01:04 · 1488 阅读 · 0 评论 -
【Pytorch】自定义的transfomer模块(torchvision)---ZeroPaddingResize
1.目的有时候直接进行resize会有形变,所以想到这样的方式,同比例缩放,然后补0。2.实现原创 2021-05-28 16:24:51 · 253 阅读 · 2 评论 -
【python】yolov5的torch与torchvision环境问题
1.可用环境torch==1.7.1+cu110torchvision==0.8.2+cu1102.错误环境遇到的问题(1)情况一torch==1.7.0+cu110torchvision==0.8.0训练yolov5时,训练就会开始报错:return torch.ops.torchvision.nms(boxes, scores, iou_threshold)RuntimeError: Could not run 'torchvision::nms' with argu原创 2021-05-15 15:50:13 · 7887 阅读 · 8 评论 -
【python】python代码中判断torch版本(用于不太版本的代码兼容)
0.环境ubuntutorch1.1.0torch1.7.0+cu1101.目的因为服务器的torch版本不一样,有些地方会无法出错。所以想到在代码中加入一个版本的判断。2.终端中实验pythonimport torchprint(torch.__version__)两台设备分别会得到:1.1.01.7.0+cu1103.代码示例import torchif torch.__version__ <= "1.1.0" pa原创 2021-03-26 11:57:32 · 2831 阅读 · 0 评论 -
torch调用模型推理结果不对分析
1.模型是否导入成功1.1 问题Missing key(s) in state_dict: "features.0.weight"Unexpected key(s) in state_dict: "module.features.0.weight"例如,如果此时使用的下面命令导入模型:net.load_state_dict(checkpoint["state_dict"], strict=False)此处strict参数设置为False(会忽略不匹配的网络层参数)。如何训练时使用的原创 2021-02-20 11:45:40 · 908 阅读 · 0 评论 -
【行人重识别】fast-reid复现(20210111)
参考代码:https://github.com/JDAI-CV/fast-reid0.环境ubuntu16.04cuda9.0python3.6torch==1.1.0torchvision==0.3.0Cythonyacstensorboardfuturetermcolorsklearntqdmopencv-python==4.1.0.25matplotlibscikit-imagenumpy==1.16.4安装apex(不要直接通过pip安装):gi原创 2021-01-12 13:54:18 · 2718 阅读 · 8 评论 -
【python】python与format输出保留两位百分比%小数位
a = 0.999999print('a is {:.2%} .'.format(a))>> a is 99.99% .原创 2021-01-11 10:54:14 · 6335 阅读 · 0 评论 -
【python】TypeError: (‘Keyword argument not understood:‘, ‘interpolation‘)
0.环境windowsscikit-learn==0.23.21.错误TypeError: ('Keyword argument not understood:', 'interpolation')2.解决版本降级pip install scikit-learn==0.22.2原创 2020-12-01 10:26:34 · 2722 阅读 · 0 评论 -
(fast-reid)torch error:DataLoader worker (pid(s) 26462) exited unexpectedly
1.问题之前的时候已经配置好了fast-reid的环境,隔了以后再次运行这个,发现出现缓存不足的问题:File "/usr/local/lib/python3.6/dist-packages/torch/multiprocessing/reductions.py", line 315, in reduce_storage fd, size = storage._share_fd_()RuntimeError: unable to write to file </torch_2745原创 2020-07-11 17:21:20 · 1379 阅读 · 0 评论 -
pytorch转caffe中maxpool一定要注意
pytorch转caffe中maxpool一定要注意,不然成功转换了还是会出错.下面给出了pytorch与caffe中maxpool层的转换的对应关系.为了自己以后不要再被这个问题坑吧.fast-reid中怎么转caffe成功且正确.pytorch layerself.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, ceil_mode=True)caffe layer:layer { name: "max_pool1" type:原创 2020-08-05 10:57:54 · 1151 阅读 · 5 评论 -
对于pytorch中dataloader PIL与opencv总结
1.PILpytorch中dataloader使用的是PIL读取数据,有用的就是下面几句:from PIL import Image, ImageOpsdef read_image(file_name, format=None): image = Image.open(file_name) image = np.asarray(image) #转为numpy的数组,只有这样才能读出这里面的数据 image = Image.fromarray(image) #转PIL,如原创 2020-08-05 11:11:45 · 1623 阅读 · 0 评论 -
pytorch转caffe的结果不一样时经验总结
最近跟转模型结果不一样斗争了很久很久.争对尝试,总结几个我们可以去查问题的方向吧.1.模型是否正确转后的模型每一层网络结构是否一样,输入与输出的特征图维度是不是一样.2.同样的输入,预处理与后处理相同,不同模型(pytorch,caffe)验证结果(1)这种情况下我们保证所有的输入/预处理/后处理一致,确定是不是模型转换中出现的问题.(2)怎么验证:每一种处理直到输入到模型,都做可视化输出,一一核对输出是否一致,不一致就改一致.3.查pytorch转caffe的一些坑,找思路(1原创 2020-08-05 11:46:03 · 3152 阅读 · 7 评论