wchh-2000 / SAMPolyBuild

Adapting the Segment Anything Model for Polygonal Building Extraction
21 stars 1 forks source link

关于SAMPoluBuild代码的问题 #1

Closed dyzy41 closed 3 weeks ago

dyzy41 commented 1 month ago

作者您好,关于您的代码我有2点问题?

  1. test.py代码中似乎没有导入训练好的模型,我猜测这里可能缺了部分代码。
  2. 请问一下这份代码做推理的时候,需要输入点或者框的prompt吗?

感谢

labiao commented 1 month ago

作者您好,关于您的代码我有2点问题?

  1. test.py代码中似乎没有导入训练好的模型,我猜测这里可能缺了部分代码。
  2. 请问一下这份代码做推理的时候,需要输入点或者框的prompt吗?

感谢

似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?

wchh-2000 commented 1 month ago

作者您好,关于您的代码我有2点问题?

  1. test.py代码中似乎没有导入训练好的模型,我猜测这里可能缺了部分代码。
  2. 请问一下这份代码做推理的时候,需要输入点或者框的prompt吗?

感谢

1.如下图,build_sam中会加载args中的训练好的权重文件checkpoint image

2.数据集标注中含有框(bbox)字段,会用bbox和bbox中心点作为提示,在dataset/dataset_crop.py ‎PromptDataset中加载。

wchh-2000 commented 1 month ago

似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?

auto mode 代码是基于RSPrompter的,我们还在整理,以后会放出

labiao commented 1 month ago

似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?

auto mode 代码是基于RSPrompter的,我们还在整理,以后会放出

好的,感谢您的工作。

labiao commented 1 month ago

作者您好,关于您的代码我有2点问题?

  1. test.py代码中似乎没有导入训练好的模型,我猜测这里可能缺了部分代码。
  2. 请问一下这份代码做推理的时候,需要输入点或者框的prompt吗?

感谢

1.如下图,build_sam中会加载args中的训练好的权重文件checkpoint image

2.数据集标注中含有框(bbox)字段,会用bbox和bbox中心点作为提示,在dataset/dataset_crop.py ‎PromptDataset中加载。

那请问下,如果想预测一张没有bbox标签的图像,代码能实现吗?prompt_fullimg配置下预测时是否需要bbox标签呢?

wchh-2000 commented 1 month ago

那请问下,如果想预测一张没有bbox标签的图像,代码能实现吗?prompt_fullimg配置下预测时是否需要bbox标签呢?

如有需要,我们可以提供推理代码,输入影像,交互式点击获取框坐标,模型预测矢量轮廓。 prompt_fullimg配置下也需要bbox标签。fullimg指整个图像输入模型,而instance配置是只有实例区域图像输入模型。

labiao commented 1 month ago

那请问下,如果想预测一张没有bbox标签的图像,代码能实现吗?prompt_fullimg配置下预测时是否需要bbox标签呢?

如有需要,我们可以提供推理代码,输入影像,交互式点击获取框坐标,模型预测矢量轮廓。 prompt_fullimg配置下也需要bbox标签。fullimg指整个图像输入模型,而instance配置是只有实例区域图像输入模型。

好的,那麻烦您提供下推理代码供我学习。非常感谢。可发送至我邮箱:1725089507@qq.com

wchh-2000 commented 1 month ago

好的,那麻烦您提供下推理代码供我学习。非常感谢。可发送至我邮箱:1725089507@qq.com

更新了代码,推理代码见infer_poly_crop.py,交互式提示推理interactive_prompt.py。

wchh-2000 commented 1 month ago

似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?

已经更新加入了自动模式的训练和测试