法治建设网站模块,阳澄湖大闸蟹网站建设,做网站发布网,长治推广型网站开发项目地址#xff1a;https://github.com/Kolkir/Coarse_LoFTR_TRT 创建时间#xff1a;2022年 相关训练数据#xff1a;BlendedMVS LoFTR [19]是一种有效的深度学习方法#xff0c;可以在图像对上寻找合适的局部特征匹配。本文报道了该方法在低计算性能和有限内存条件下的…项目地址https://github.com/Kolkir/Coarse_LoFTR_TRT 创建时间2022年 相关训练数据BlendedMVS LoFTR [19]是一种有效的深度学习方法可以在图像对上寻找合适的局部特征匹配。本文报道了该方法在低计算性能和有限内存条件下的设备上的优化工作。原来的LoFTR方法是基于一个ResNet [6]backbone和两个基于线性transformer[22]架构的模块。在本研究中只剩下粗匹配块参数的数量显著减少并使用知识蒸馏技术对网络进行了训练。对比结果表明在粗匹配块中尽管模型大小显著减少但该方法仍可以获得适当的特征检测精度。此外本文还展示了使模型与NVIDIA TensorRT运行时兼容所需的额外步骤并展示了一种优化针对低端gpu的训练方法的方法。
简化后的算法运行速度在1060显卡上提升了45倍针对640×480图像fps可达45
1、改进思考
1.1 算法背景
为了解决高计算复杂度的问题对transformer架构的各种修改已经被开发出来。LoFTR方法采用线性变换[22]方法该方法提出通过将注意层中使用的指数核替换为··1.的替代核* ( K ) T (K)^T (K)T从而降低计算复杂度到该方法对计算机视觉任务具有良好的计算性能的提高和内存消耗的降低。这很重要因为这种类型的任务的序列长度等于输入图像中的像素数。使用线性transformer允许对640x480张的图像进行特征匹配在高端gpu上具有可接受的性能。然而该体系结构中的更改仍然不足使transformer可以运行在低端gpu上。
1.2 优化方向
使复杂模型适应低端器件[5]要求的工程方法主要有量化、剪枝[9]和知识蒸馏。
量化是用于计算和权重存储的数据类型的位宽降低。通常浮点计算转换为16位浮点或8位整数类型。为了达到与原始模型相媲美的精度这种方法通常需要一个特殊的训练过程考虑到缩小或缩小后的附加模型校准。量化通常在消费者级或嵌入式gpu上不可用而且它的实现只能在高端gpu中可用。然而对于基于cpu的设备该方法是可用的可以提供良好的效果。
剪枝是一种去除网络参数的方法它对结果的精度没有太大的贡献。通常一个合适的剪枝条件是权重接近于零。由此得到的模型可能需要更少的内存而且在推理方面可能更有效。有许多剪枝类型但可以区分以下两种主要类型结构化剪枝当对称的权值块被删除时例如层和非结构化剪枝当被删除的块可能是不同的形状时。由于这种方法改变了模型架构因此通常需要进行手动调整来恢复正常的模型工作。结构化方法可能更可取因为它对全局架构进行的更改更少而且恢复模型操作更容易甚至可能不必要。然而流行的深度学习框架通常会实现非结构化的方法。在复杂模型中应用非结构化处理后适应网络操作可能是一项重要的任务需要很多时间来解决而且由于该方法不能保证一个稳定的结果因此应用它并不总是合适的。
知识蒸馏是在教师的帮助下训练模型的一种方法。教师可以是具有相同架构但具有更多参数的网络也可以是具有其他架构的网络。大多数训练是使用复杂的损失函数转移教师的知识。转移到学生模型中的知识元素可以是教师网络中某些层的输出值例如在分类中可能是softmax之前的输出。也可以使用教师网络[2]的内部层输出值。知识蒸馏在保持所需的精度的同时显示了良好的结果但没有标准的方法来组织这样的过程。而成功则取决于正确的知识转移技术、精心选择的损失函数和学生模型架构。
如上所示没有单一的方法来优化低端设备的深度学习模型。因此通常会针对特定的架构开发专门的解决方案。本文提出了一种针对LoFTR特征匹配方法的优化方法。
1.3 本文方案
该方法的主要思想是显著减少模型参数的数量和从原始模型中的知识转移。 决定只保留一个transformer block用于粗特征匹配尽管原始模型包含第二个transformer模块用于细匹配同时在所有模型块中进行了手动迭代选择较少的层网络结构简化。设计了知识蒸馏损失函数并使用了一个较小的训练数据集设计知识蒸馏方案。然而地面-真实的特征点的匹配也可以使用深度图来确定。训练过程是开发使用自动混合精度AMP技术和梯度积累方法来节省内存和加快计算。
源代码被改编为以NVIDIA TensorRT [13]引擎格式编译。选择工作内存大小为2Gb的NVIDIA Jetson Nano [12]作为目标设备。并选择了基于英特尔i5处理器和Nvidia GTX 1060 6Gb GPU的桌面机作为训练平台。
2、模型改进
2.1 适配性修改
最初的LoFTR模型是用Python编写的使用PyTorch作为深度学习框架。为了创建TensorTR模型有两种可能性一种是使用torch-TensorRT[14]编译器第二种是将模型转换为ONNX [1]格式然后使用NVIDIA TensorTR SDK编译它。由于目标平台的资源有限无法应用第一个选项因为使用Torch-TensorRT编译成TensorTR格式意味着在目标设备上运行它以进行实时优化。实验发现编译ONNX需要的资源更少并且在目标设备上是可能的因此选择了第二个选项。
然而einsum操作在onnx中并不支持。 所有将运算方式修改为以下使onnx与tensorRT都支持。
2.2 结构优化
为了目标设备上实现可接受的性能即选择块中的层的数量和尺寸。为此目的我们开发了一个在实时网络摄像头图像上搜索特征匹配的演示应用程序。性能是通过呈现相应匹配时的FPS数量来估计的。然后在此应用程序的帮助下迭代地选择了表1中所示的模型配置。
原始模型的作者报告说完整模型在RTX 2080Ti上处理116 ms处理一对640×480图像约8 FPS [19]。简化后的算法运行速度在1060显卡上提升了45倍。 表3显示了参数数量的变化。从表中可以看到原始模型的尺寸显著减少以便在目标设备上实现可接受的性能。
2.3 训练设置
针对低性能硬件的局限性对知识精馏训练过程进行了优化。为了加速梯度计算和减少内存消耗我们使用了自动混合精度AMP技术因为它的实现在PyTorch深度学习框架中可用。该技术的本质是梯度计算所需的一些操作使用浮点32而另一部分使用浮点16种数据类型。例如卷积运算和线性层相关的矩阵计算使用float16计算速度更快。而其他操作如减法需要使用一个浮动32范围。这项技术使我们能够为模型训练中涉及的所有操作自动选择适当的数据类型。它的使用可以显著减少模型ResNetFPN头的内存消耗。然而AMP技术存在较小梯度值的数值计算问题。因此为了稳定损失函数增加了放大因子。
尽管使用了AMP但在GTX 1060上进行训练也只能是支持到batch为4的640x480的图像。因此为了增加batch的大小我们采用了梯度积累的方法。这意味着大batchsize被分为系列的小batchsize。对于每个系列进行正向和反向循环不清除产生的梯度值而是求和。其中ℎ、ℎ。在每次迭代中损失函数值乘以比例因子1/。只有经过所有迭代后才更新网络参数然后将梯度归零。因此利用该技术模拟了大批量的训练。在这项工作中虚拟批处理大小等于32。尽管在现实中硬件处理了8批梯度累计每批的尺寸为4。梯度积累技术并没有实现实际大批量使用的精确对应关系因此这两种方法的损失和梯度值将是不同的。
此外我们还注意到应用学习速率调度器可以显著加快训练过程。本研究采用了具有标准参数的AdamW [10]优化算法。初始学习速率值为10−3每15个epoch乘以10−3。
每个epoch都从原始数据集中随机选择大小为5000对图像。
2.4 训练效果
图1显示了有教师和没有教师的训练的损失函数值。这张图清楚地表明当与教师一起进行训练时绝对损失函数值明显更小学习过程本身更稳定。 图2它显示了平均绝对误差MAE与训练持续时间的依赖关系。它显示了预测的特征匹配分数与地面真实值之间的平均差异。我们可以看到当与老师一起进行训练时MAE值远远接近于零。我们可以假设在没有教师的情况下训练一个较小的网络会使它对其结果缺乏信心。然而与此同时这个图1显示了所选择的模型架构能够在没有老师的情况下学习但可能需要更长的时间来获得可比的结果并且需要更低的阈值来确定最重要的匹配。 图3显示了在数据集图像上的模型结果的示例。白点表示原模型作为教师使用的粗LoFTR模块的匹配结果。黑点表示较小模型的结果。从实验结果中可以清楚地看出较小的模型比教师模型更关注图像的不同部分。最可能的原因是头层数量较少transformer参数不同使得模型强调更明显的特征点。也可以注意到较小模型的特征匹配中存在错误尽管通常特征匹配是相当准确的。 室外数据配准效果
3、代码运行
打开 https://github.com/Kolkir/Coarse_LoFTR_TRT即可下载项目
3.1 前置修改
如果电脑没有摄像头则需要进行下列额外代码修改
修改一 webcam.py中默认参数camid类型修改为str默认值修改为自己准备好的视频文件
def main():parser argparse.ArgumentParser(descriptionLoFTR demo.)parser.add_argument(--weights, typestr, defaultweights/outdoor_ds.ckpt,helpPath to network weights.)# parser.add_argument(--camid, typeint, default0,# helpOpenCV webcam video capture ID, usually 0 or 1.)parser.add_argument(--camid, typestr, defaultrC:\Users\Administrator\Videos\风景视频素材分享_202477135455.mp4,helpOpenCV webcam video capture ID, usually 0 or 1.)修改二camera.py中的代码修改为以下用于支持读取视频文件
import cv2
from threading import Threadclass Camera(object):def __init__(self, index):self.indexindexif isinstance(self.index,int):#加载摄像头视频流self.cap cv2.VideoCapture(self.index, cv2.CAP_V4L2)else:#加载视频self.cap cv2.VideoCapture(self.index)if not self.cap.isOpened():print(Failed to open camera {0}.format(index))exit(-1)# self.cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)# self.cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)self.thread Thread(targetself.update, args())self.thread.daemon Trueself.thread.start()self.status Falseself.frame Nonedef update(self):while True:try:if self.cap.isOpened():(self.status, self.frame) self.cap.read()if not self.status:if isinstance(self.index,int):#加载摄像头视频流self.cap cv2.VideoCapture(self.index, cv2.CAP_V4L2)else:#加载视频self.cap cv2.VideoCapture(self.index)else:breakexcept cv2.error as e:print(e)breakdef get_frame(self):return self.frame, self.statusdef close(self):self.cap.release()self.thread.join()
3.2 运行效果
然后运行webcam.py可以发现fps为25左右此时硬件环境为win10笔记本、1660显卡26ms即可处理完一个640*480的图片。但整体fps稳定在16~26左右。 再次加速将推理时的图像分辨率修改为320x240 即将webcam.py中的 img_size 设置(320, 240)loftr\utils\cvpr_ds_config.py中对应的设置。发现速度没有显著提升但整体fps稳定在22~28左右。
_CN.INPUT_WIDTH 320
_CN.INPUT_HEIGHT 240onnx运行效果如下整体fps稳定在20左右 将模型配置loftr\utils\cvpr_ds_config.py 中的尺寸修改如下然后重新运行export_onnx.py导出模型再基于webcam.py运行onnx模型可以发现fps高达40以上。
_CN.INPUT_WIDTH 320
_CN.INPUT_HEIGHT 3203.3 图像配准
使用Coarse_LoFTR_TRT进行图像配准可以参考 https://blog.csdn.net/a486259/article/details/140241276 中章节5的操作。操作前最好先修改 loftr\utils\cvpr_ds_config.py 的尺寸为320具体修改如下然后重新运行export_onnx.py导出模型。
_CN.INPUT_WIDTH 320
_CN.INPUT_HEIGHT 320
