Poor performance in a object with two colors.

nv-nguyen / cnos

[ICCV 2023 R6D] PyTorch implementation of CNOS: A Strong Baseline for CAD-based Novel Object Segmentation based on Segmenting Anything and DINOv2

MIT License

191 stars 21 forks source link

Poor performance in a object with two colors. #21

Closed liuchangji closed 3 months ago

liuchangji commented 3 months ago

First of all, thank you for sharing such an outstanding piece of work. I have an issue here. I have a pair of pliers where you can observe two distinct colors on the upper and lower parts. Consequently, CNOS is only segmenting the yellow section on top. Do you have any suggestions to address this problem? I believe this might be an inherent limitation of the SAM model.

img_v3_02b5_60e7a8e8-cf1a-43da-8270-8148c7fd166g

qidihan commented 3 weeks ago

Hi, did you solve this problem? I'm having the same issues. It seems the problem of SAM model, I tried to change the stability and iou score on SAM model, still didn't fix this problem.