PSWNet
三维视觉工作室
深度学习
展开
-
PSMNet训练中遇到的报错
1.main.py是训练10 epochs的sceneflow的例子。2.finetune.py是训练300 epochs的KITTI的例子。只跑了这个例子首先修改了 finetune.py的路径,把KITTI路径和预训练模型加载进来。parser.add_argument('--datapath', default='./data_scene_flow_2015/training/', help='datapath')parser.add_argumen原创 2020-11-06 18:42:52 · 1849 阅读 · 11 评论 -
立体匹配 -- PSM-Net 网络模型代码剖析
只熟悉流程跑通代码不重要,重要的是理解网络的思想。GC-Net提出了3D-CNN编解码的形式做’cost volum ’ 后处理的过程,PSM-Net 加入图像金字塔的模块结合3D-CNN 输出图像视差图。一. 特征提取模块作者用 3层 33的小卷积核代替 77 的大卷积核,将图像降维1/2size. 虽然拥有同样大小的感受野,但深层的小 conv filter 显然有更少的参数,降低了计算成本。def convbn(in_planes, out_planes, kernel_size.原创 2020-11-02 17:03:10 · 4479 阅读 · 11 评论 -
《Pyramid Stereo Matching Network》--PSMNet 测试代码实现 +如何解决GPU内存不足问题
导师给了建议,转战DL方向做立体匹配了。PSWNet 是cvpr2018的文章,立体匹配的经典,下面介绍一下我是怎么跑通文章的测试代码,以及踩过的坑。-对于复现论文,我也是初学者,希望给更多刚入门的师弟师妹一些帮助。一.下载github源码1.一般论文都是有github连接的2.如何阅读github以下就是github中的目录注意到 dataloader 文件夹是针对数据输入预处理的,其实就是怎么 crop 图像,怎么对图像做 BatchNormalizemodels 文件夹就是.原创 2020-10-24 21:32:01 · 3473 阅读 · 8 评论