摘要: #问题 训练网络往往需要加载预训练模型,主流的就是ResNet一类的预训练好的参数 但我在加载了预训练模型,并冻结与训练参数后,进行训练时,发现了两个问题 ##1 在进行test中model.train()的准确率要远高于model.eval()差别大概在7个点左右。 其中model.eval() 阅读全文
posted @ 2022-06-20 00:56 The1912 阅读(1605) 评论(0) 推荐(0) 编辑
摘要: #问题 感谢pycharm,我还不知道有一天我会发现这种问题,我本来是查看一下batchnorm2d中tensor的requires_grad属性,然后我就顺着快捷栏点下来的。结果发现requires_grad_()与requires_grad完全不一样。 #代码 requires_grad for 阅读全文
posted @ 2022-06-20 00:32 The1912 阅读(777) 评论(0) 推荐(0) 编辑
摘要: #目标 在训练网络的时候,往往需要实现一个实时输出epoch、batch_id、loss的方法。当然我们可以一个epoch输出一次,但为了更好的了解loss的变化,我们可以每次迭代都输出一次。但每次print都会重启一行,这样一个epoch下来会输出很多行,非常的不方便。我们可以巧用print函数来 阅读全文
posted @ 2022-06-16 18:51 The1912 阅读(191) 评论(0) 推荐(0) 编辑
摘要: #问题 问题是这样的,我简单描述一下。我在处理NYU标签图的过程中被imwrite实实在在坑了一把。我处理的是40classes的标签图,其中从matlab中读出的矩阵格式是uint16 然后我准备通过imwrite的方式将其保存为png图像 cv2.imwrite(dir, labels[:, : 阅读全文
posted @ 2022-06-13 21:08 The1912 阅读(383) 评论(0) 推荐(0) 编辑
摘要: #内容 也就是稍微mark一下。平时的深度相机所采集的数据一般是一个三通道的RGB图像和一个单通道的depth图像(深度信息),但其实为了更好的让单通道的深度信息被CNN所处理,提出了基于depth的HHA编码方法。详细内容可以看这篇论文,有机会的话我更新在论文集合里,只能说大佬就是大佬,能想到这些 阅读全文
posted @ 2022-06-13 00:06 The1912 阅读(840) 评论(0) 推荐(0) 编辑
摘要: #记录 算是自己一点点小小的记录,以前很少看开源的代码,都自己闷头写,最后才发现自己写的就是shi。不看不学不练啊,读开源代码不代表不自己造轮子,而是要学会别人编程的思想并学习,自己检讨我自己。 最近确实压力很大,想把这篇文章水出来,可是一来没有人带,导师是拉项目大师,只会分配杂活,教你是不可能的, 阅读全文
posted @ 2022-06-11 23:21 The1912 阅读(496) 评论(0) 推荐(0) 编辑
摘要: #问题 关于在pytorch中冻结参数,我看网上大多都使用了 for param in net.parameters(): param.requires_grad = False 但我的需求是这样的,我创建了一个dict,里面含有对应的tensor的name与param,我想让根据这些名称对网络的参 阅读全文
posted @ 2022-06-10 23:14 The1912 阅读(1614) 评论(0) 推荐(0) 编辑
摘要: #1 首先,加载是有条件的,就是两个模型想要加载的参数的名字相同,才能加载进来。 #2 加载的方法有两种,load_state_dict(strict)和update 代码里的方法是对backbone单独做一个Module类,这样不容易出错。 代码中展示了如何把train_net中的backbone 阅读全文
posted @ 2022-05-25 23:16 The1912 阅读(984) 评论(0) 推荐(0) 编辑
摘要: #1 由于之前一直没有用过torch中自带的可视化工具 torchsummary,就安装了试一下 好像用conda无法安装,只能用pip安装。 进入相应的conda虚拟环境输入 pip install torchsummary 注意别忘了关vpn或者代理之类的,不然会报错 #2 起因是这样,安装了后 阅读全文
posted @ 2022-05-25 20:51 The1912 阅读(707) 评论(0) 推荐(0) 编辑
摘要: #1 本来我以为用pycharm封装好的包就可以随意调动了,然而放到服务器上没有pycharm时发现,python并不能自动读取非当前文件夹下的别的路径下的库或类,可能是因为pycharm比较智能化吧,有工程文件可以引用。 #2 解决办法 来源于[1] 同一文件夹 import file1 导入下级 阅读全文
posted @ 2022-05-13 10:29 The1912 阅读(736) 评论(0) 推荐(0) 编辑