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