Closed dyzy41 closed 3 weeks ago
作者您好,关于您的代码我有2点问题?
- test.py代码中似乎没有导入训练好的模型,我猜测这里可能缺了部分代码。
- 请问一下这份代码做推理的时候,需要输入点或者框的prompt吗?
感谢
似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?
作者您好,关于您的代码我有2点问题?
- test.py代码中似乎没有导入训练好的模型,我猜测这里可能缺了部分代码。
- 请问一下这份代码做推理的时候,需要输入点或者框的prompt吗?
感谢
1.如下图,build_sam中会加载args中的训练好的权重文件checkpoint
2.数据集标注中含有框(bbox)字段,会用bbox和bbox中心点作为提示,在dataset/dataset_crop.py PromptDataset中加载。
似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?
auto mode 代码是基于RSPrompter的,我们还在整理,以后会放出
似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?
auto mode 代码是基于RSPrompter的,我们还在整理,以后会放出
好的,感谢您的工作。
作者您好,关于您的代码我有2点问题?
- test.py代码中似乎没有导入训练好的模型,我猜测这里可能缺了部分代码。
- 请问一下这份代码做推理的时候,需要输入点或者框的prompt吗?
感谢
1.如下图,build_sam中会加载args中的训练好的权重文件checkpoint
2.数据集标注中含有框(bbox)字段,会用bbox和bbox中心点作为提示,在dataset/dataset_crop.py PromptDataset中加载。
那请问下,如果想预测一张没有bbox标签的图像,代码能实现吗?prompt_fullimg配置下预测时是否需要bbox标签呢?
那请问下,如果想预测一张没有bbox标签的图像,代码能实现吗?prompt_fullimg配置下预测时是否需要bbox标签呢?
如有需要,我们可以提供推理代码,输入影像,交互式点击获取框坐标,模型预测矢量轮廓。 prompt_fullimg配置下也需要bbox标签。fullimg指整个图像输入模型,而instance配置是只有实例区域图像输入模型。
那请问下,如果想预测一张没有bbox标签的图像,代码能实现吗?prompt_fullimg配置下预测时是否需要bbox标签呢?
如有需要,我们可以提供推理代码,输入影像,交互式点击获取框坐标,模型预测矢量轮廓。 prompt_fullimg配置下也需要bbox标签。fullimg指整个图像输入模型,而instance配置是只有实例区域图像输入模型。
好的,那麻烦您提供下推理代码供我学习。非常感谢。可发送至我邮箱:1725089507@qq.com
好的,那麻烦您提供下推理代码供我学习。非常感谢。可发送至我邮箱:1725089507@qq.com
更新了代码,推理代码见infer_poly_crop.py,交互式提示推理interactive_prompt.py。
似乎还有一些问题,configs文件中提供的配置文件都是基于prompt mode的,关于auto mode 的配置文件好像没有提供。另外,模型部分是否有提供 auto bbox prompt 模块相关的代码?
已经更新加入了自动模式的训练和测试
作者您好,关于您的代码我有2点问题?
感谢