Jumpat / SegmentAnythingin3D

Segment Anything in 3D with NeRFs (NeurIPS 2023)
Apache License 2.0
831 stars 52 forks source link

SAM模型使用 #66

Open tourbliu opened 2 months ago

tourbliu commented 2 months ago

您好,我在使用您的代码进行训练时候,是能够训练,但我将sam_vit模型换成别的sam模型之后,在dash不能够选择点,但我已经将对应的segment anything文件夹也进行更换并重新进行编译了,我查看代码的时候,点的选择似乎与sam模型是无关的,请问这是什么原因呢? 我查看终端,出现的报错是没mask的生成问题,我想主要问题还是在点的选择? image

Zanue commented 1 month ago

您好,看报错似乎是mask并没有生成。

将sam_vit模型换成别的sam模型之后,在dash不能够选择点

请问没有更换sam_vit模型的时候,代码训练是正常的么?是的话,您是如何更换sam_vit模型的呢?

tourbliu commented 1 month ago

您好,非常感谢您的回复。在没有更换sam_vit模型的时候是能够正常训练的。我是将sam_vit预训练进行更换,以及重新下载对应的segment_anything文件夹进行重新编译,代码是修改了sam3d.py中的 image 其他的未进行修改。