由于原始训练代码无法正常训练,在github中找到了别人写的代码,经测试,可以正常训练。
代码连接
代码修改:(只需要修改图片输入路径即可)
训练UnsuperPoint
修改UnsuperPoint_coco.yaml中的train_path
训练UnsuperPoint
修改coco.py文件中self.is_training的base_path(训练图片路径,加上之前修改的train_path,组成完整的训练图片存放路径)。
修改coco.py文件中else的base_path(评估图片路径,加上之前修改的train_path,组成完整的评估图片存放路径)。

运行指令 python train.py

相关文章:

  • 2021-04-30
  • 2021-06-16
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-02-19
  • 2022-12-23
  • 2021-11-11
猜你喜欢
  • 2021-08-23
  • 2021-06-02
  • 2021-10-15
  • 2022-02-19
  • 2021-12-23
  • 2021-08-25
  • 2021-07-29
相关资源
相似解决方案