深度学习
Daniel@1994
这个作者很懒,什么都没留下…
展开
-
Warning: masked_scatter_ received a mask with dtype torch.uint8, this behavior is now deprecated,ple
Warning: masked_scatter_ received a mask with dtype torch.uint8, this behavior is now deprecated,please use a mask with dtype torch.bool instead 查看代码中加载训练数据集的地方,在生成mask标签的函数中将return中mask换为mask.bool()即可: heatmaps, mask = generate_label_map(Hpoint, heig原创 2021-01-14 11:41:42 · 2706 阅读 · 0 评论 -
pytorch 单gpu改为多gpu训练和测试
pytorch 单gpu改为多gpu训练和测试 训练中 在plfd_backbone = plfd_backbone.to(device)之前加 import torch.nn as nn plfd_backbone = nn.DataParallel(plfd_backbone, device_ids = [0, 1, 2, 3]) device_ids:设置gpu个数。 测试时,同样要修改 在to(device)后加入 pfld_backbone = nn.DataParallel(pfld_ba原创 2020-12-29 11:39:54 · 2357 阅读 · 1 评论 -
win10下PaddleOCR初体验
PaddleOCR初体验原创 2020-11-09 14:47:04 · 1825 阅读 · 0 评论 -
win10下 pip安装dlib
win10下 pip安装dlib 安装步骤: 1.安装visual studio2017 2.pip 安装 Cmake 3.pip 安装 dlib 问题描述: dlib库需要C++编译,因此要安装C++编译环境。 解决方案: 1.安装visual studio2017,如图 2.pip install cmake (如果下载慢,请换清华源下载) 3.pip install dlib ...原创 2020-10-20 14:26:15 · 999 阅读 · 0 评论 -
python批量修改xml文件内容
由于cityperson数据集图片较大,因此将图片裁剪成4份,同时对应修改标注信息 修改前的xml文件 aachen_000000_000019_leftImg8bit.xml 在这里插入代码片 #python import xml.etree.ElementTree as ET import numpy as np import os '''由于cityperson行人数据集图片较大,将图片裁剪成4份,同时对应修改标注信息其中左下裁剪图片代码如下,其他位置对应修改裁剪坐标即可,同时修改标注信息。'''原创 2020-07-06 12:53:32 · 1961 阅读 · 0 评论 -
py-rfcn中遇到的错误
eltwise_layer.cpp:34 check failed: bottom[i]->shape() == bottom[0]->shape(). error) 这个错误是在执行element-wise(concat或sum)时产生的,此使只需要根据日志,查找到相加的两个元素,对其维度调整,基本就可以解决问题。 训练py-R-FCNwithFPN过程中报错 在py-rfcn中加入FPN网络结构,产生如下错误:(错误链接: link.) 错误原因:FPN网络模型中的下采样操作使得特征图分辨原创 2020-06-30 11:56:42 · 1202 阅读 · 2 评论