Advertisement

SuperPoint-Pytorch：基于HTTPS的SuperPoint PyTorch实现

  •  5星
  •     浏览量: 0
  •     大小:None
  •      文件类型：None

简介：
SuperPoint-Pytorch 是一个基于HTTPS的项目，提供了用PyTorch框架实现的SuperPoint算法代码。此项目便于研究者和开发者在图像特征检测任务中使用与改进该模型。 超点火炬是Superpoint模型的PyTorch实现和评估文件。我们在Rémi Pautrat的TensorFlow实现中得到了很大的帮助。 在兴趣点检测方面，我们的模型似乎没有完全收敛，但与同形加法结合使用时结果看起来不错。 与其他点检测模型相比，虽然总体效果不如原始模型，但在匹配点的数量上有所差异：对于原始模型而言是这样的情况；而在我们实施的版本中则是另一番景象。尽管目前的整体表现不尽如人意，但我们希望将来能够利用不同的模块（例如数据生成、单应性调整等）进行改进。 该文件涵盖了实现的所有阶段： 1. 生成综合数据集 - 创建一个包含100,000个人造合成形状图像的数据集，并附带名称和标签的相应文件。此步骤在Tesla V-100上大约需要耗时12小时。 2. 使用合成数据集进行Magicpoint训练。

全部评论 (0)

还没有任何评论哟~
客服
客服
客服关闭
  • SuperPoint-Pytorch：基于HTTPS的SuperPoint PyTorch实现
    优质
    SuperPoint-Pytorch 是一个基于HTTPS的项目，提供了用PyTorch框架实现的SuperPoint算法代码。此项目便于研究者和开发者在图像特征检测任务中使用与改进该模型。 超点火炬是Superpoint模型的PyTorch实现和评估文件。我们在Rémi Pautrat的TensorFlow实现中得到了很大的帮助。 在兴趣点检测方面，我们的模型似乎没有完全收敛，但与同形加法结合使用时结果看起来不错。 与其他点检测模型相比，虽然总体效果不如原始模型，但在匹配点的数量上有所差异：对于原始模型而言是这样的情况；而在我们实施的版本中则是另一番景象。尽管目前的整体表现不尽如人意，但我们希望将来能够利用不同的模块（例如数据生成、单应性调整等）进行改进。 该文件涵盖了实现的所有阶段： 1. 生成综合数据集 - 创建一个包含100,000个人造合成形状图像的数据集，并附带名称和标签的相应文件。此步骤在Tesla V-100上大约需要耗时12小时。 2. 使用合成数据集进行Magicpoint训练。
     
  • 利用pytorch-superpoint和pytorch-superglue进行自定义数据集的训练
    优质
    本项目旨在使用PyTorch框架下的SuperPoint与SuperGlue模型，针对特定视觉任务优化，并基于用户定制的数据集开展深度学习训练。通过调整模型参数及采用创新性损失函数，以提升特征匹配精度和鲁棒性，在计算机视觉领域如图像检索、物体识别等方面展现应用潜力。 SuperPoint与SuperGlue的组合可以实现基于深度学习的图像配准。官方发布的SuperPoint与SuperGlue模型均是基于COCO数据集进行训练的，这可能与业务中的实际数据存在一定的差距。因此，我们希望通过开源项目pytorch-superpoint和pytorch-superglue来训练自己的数据集，并优化pytorch-superpoint在训练过程中的诸多细节问题。本段落档将详细介绍如何使用这两个项目来进行图像配准模型的实验性训练。 训练完成后，为了部署这些模型，可以参考相关的技术文档或资源进行调整（支持将模型导出为ONNX格式以实现部署）。
     
  • RandLA-Net-pytorch: 基于HTTPS的RandLA-Net PyTorch实现
    优质
    简介：RandLA-Net-pytorch是基于HTTPS的高效点云特征学习库，提供PyTorch版本的RandLA-Net深度网络模型，适用于大规模点云语义分割任务。 RandLA-Net-pytorch 存储库包含该实现的代码，并且目前仅支持SemanticKITTI数据集。我们欢迎更多的人加入共同开发以提高项目的知名度。 我们的模型性能与原始TensorFlow版本相近，验证集中我们的mIoU为52.9%，而原版为53.1%。 我们在目录中提供了预训练模型，并在验证集（seq 08）上展示了结果对比： - 原始实现的mIoU：0.531 - 我们的Pytorch版本的mIoU：0.529 具体各类别的详细表现如下： 车 - 0.919，自行车 - 0.122，摩托车 - 0.290，卡车 - 0.660，其他车辆 - 0.444， 人骑自行车的人 - 0.515，电单车司机 - 0.676，路 - 0.000，停车处 - 0.912， 人行道 - 0.421，其它地面建造栅栏植被树干地形极交通标志行人 - 0.759， 其他（未明确分类）- 0.354
     
  • SuperPoint-Pretrained-Network.zip
    优质
    SuperPoint-Pretrained-Network.zip包含了预先训练好的SuperPoint模型，该模型用于检测图像中的关键点和描述子。适用于快速部署在各类视觉任务上。 SuperPointPretrainedNetwork 用于 SLAM 前端（包含 KLT 跟踪）在低纹理环境下运行的 Python 实现可以直接运行且无需改动。数据文件也在该仓库中提供。
     
  • PyTorch-UNet: PyTorch中的UNet模型实现-https
    优质
    PyTorch-UNet是一款在PyTorch框架下实现的经典卷积神经网络模型UNet的开源项目。它专为图像分割任务设计，提供高效且灵活的代码结构，适用于医疗影像分析等多种应用场景。 U-Net-PyTorch实施 这是一种流行的图像分割网络的实现方式。它非常稳定且可配置，并已在多个数据集中使用，作为几个项目的一部分。 更新：现在支持基于3-D卷积的分段。 更新：所有批次归一化操作已被实例归一化所取代（以解决医学图像中的小批量问题），并且ReLU激活函数被替换为LeakyReLU，因为它在最近的工作中得到了更广泛的应用。 安装 您可以将此软件包安装到本地Python环境中，并将其作为模块导入项目中。 步骤如下：首先克隆此存储库至您选择的文件夹内。然后进入该目录并按照以下命令安装依赖项： ``` cd pip install -r requirements.txt ```
     
  • SiamRPN-PyTorch: 基于PyTorch的SiamRPN实现
    优质
    SiamRPN-PyTorch是一款基于PyTorch框架开发的代码库，实现了Siamese Region Proposal Network（SiamRPN）算法。该工具为视觉追踪任务提供了高效、灵活的解决方案。 SiamRPN-PyTorch 使用 PyTorch 框架为对象跟踪应用程序重新实现了 SiamRPN。开发的代码基于先前的一些实现工作，并且为了测试所开发的代码，使用了 VOT-2013 体操子数据集。要求 Python 版本 >= 3.6、PyTorch 版本 >= 1.0.1、torchvision 和 cv2 库。 训练和追踪：已实现的代码以 [x1, y1, w, h] 的形式接收地面真实值（ground truth）。数据集结构如下： ``` dataset_path/Gymnastics/img1.jpg /img2.jpg ... /imgN.jpg /groundtruth.txt ``` 运行命令： $ python3 SiamRPN_train.py
     
  • EfficientNet-PyTorch：基于PyTorch的EfficientNet实现
    优质
    EfficientNet-PyTorch是使用PyTorch框架开发的EfficientNet模型的高效实现，适用于图像分类任务。它通过自动模型缩放策略优化了网络结构和参数大小。 使用EfficientNet PyTorch可以通过pip install efficientnet_pytorch命令安装，并通过以下代码加载预训练的模型： ```python from efficientnet_pytorch import EfficientNet model = EfficientNet.from_pretrained(efficientnet-b0) ``` 更新记录如下： - 2020年8月25日：新增了一个`include_top（默认为True）`选项，同时提高了代码质量和修复了相关问题。 - 2020年5月14日：增加了全面的注释和文档支持（感谢@workingcoder贡献）。 - 2020年1月23日：基于对抗训练添加了新的预训练模型类别，名为advprop。
     
  • SMOTE-Pytorch：基于Pytorch的SMOTE实现
    优质
    SMOTE-Pytorch 是一个利用 PyTorch 框架高效实现 SMOTE (Synthetic Minority Over-sampling Technique) 算法的项目，旨在解决类别不平衡问题，适用于深度学习任务的数据预处理。 当分类标签的分布不均衡时，数据集就会出现不平衡状态，在诸如欺诈检测之类的大量现实问题中，这种不平衡往往达到100比1的程度。尽管已经采取了多种方法来解决这一难题，但这个问题仍然备受关注，并且是研究的一个活跃领域。这里展示的是SMOTE（综合少数族裔过采样技术）的Pytorch实现版本。 关于SMOTE算法的相关内容和原理可以参考其原始论文《SMOTE: Synthetic Minority Over-sampling Technique》。
     
  • ERNIE-Pytorch：基于Pytorch的ERNIE实现
    优质
    ERNIE-Pytorch是一个基于PyTorch框架构建的开源项目，它实现了百度的预训练语言模型ERNIE，为自然语言处理任务提供强大的工具支持。 ERNIE是基于Bert模型构建的，在中文自然语言处理任务上表现出色。 您可以采用以下三种方式来使用这些强大的模型： 直接加载（推荐） 以ernie-1.0为例： ```python from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained(nghuyong/ernie-1.0) model = AutoModel.from_pretrained(nghuyong/ernie-1.0) ``` 您可以访问Hugging Face的模型中心以查找所有支持的ERNIE版本。 以下是几个模型及其在Transformer中的标识符： | 模型 | Transformer中的标识符 | 描述 | |-----------|----------------------|----------| | ernie-1.0（中文） | nghuyong/ernie-1.0 | 中文版ERNIE，性能优越 | 下载模型的权重文件可以通过访问相应的链接来完成。
     
  • Flownet2-Pytorch：基于Pytorch的FlowNet 2.0实现
    优质
    Flownet2-Pytorch是利用PyTorch框架重现的FlowNet 2.0项目。该项目提供了一个灵活且高效的平台，用于学习和研究光流估计技术。 Flownet2-pytorch是FlowNet的PyTorch实现版本。它支持多GPU训练，并提供有关干净数据集与最终数据集的训练或推理示例。相同的命令可用于其他数据集的训练或推断，详情请参考相关文档。此外，该库还支持使用fp16（半精度）进行推理。 网络架构部分提供了多种不同的Flownet神经网络结构选项：FlowNet2S、FlowNet2C、FlowNet2CS、FlowNet2CSS 和 FlowNet2SD。每个网络的BatchNorm版本也可用。需要注意的是，FlowNet2或其衍生模型（如FlowNet2C*）依赖于自定义层Resample2d和Correlation 。这些定制层与CUDA内核的PyTorch实现可供使用。 数据加载器部分提供了相关说明和支持。