
SuperGlue-pytorch的培训。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
SuperGlue-pytorch培训的改编工作,源于:1、它建立在官方超点技术的底层实现之上;2、该培训方案支持训练批次大小大于1的场景;3、通过优化损失函数,训练速度提升了十倍;4、此外,还启用了训练集中的非线性扭曲功能;5、同时,它包含训练集中存在的负对齐机制;6、为了增强训练效果,该培训方案推荐使用离线数据生成方式构建训练集,具体命令为:`python -m dataset.data_builder --debug 1` 进行训练,并使用 `python train.py --train_path {train_path} --superpoint_weight ./models/weights/superpoint_v1.pth --feature_dim 256 --dataset_offline_rebuild 1 --pretrained ` 参数进行设置,批次大小设置为32,并开启调试模式。SuperGlue PyTorch完整文档已提供 PDF 格式。该项目依赖 Python 3 环境。
全部评论 (0)
还没有任何评论哟~


