SAM2-UNet

SAM2-UNet copied to clipboard

calculate evaluation metrics such as Dice, IOU, and HD

5

Dear AI Engineer, how can I calculate evaluation metrics such as Dice, IOU, and HD? Could you provide me with some code? I found some code online regarding this topic,...

hczyni

Salinet Object Detection

1

How is it different from rgb salient object detection models like birefnet, inspyrenet: https://paperswithcode.com/sota/salient-object-detection-on-dut-omron. Do these models have better accuracy than SAM2-UNet for general purpose usage?

anilsathyan7

question

2

Does this mean fine-tuning only on the training dataset of DUTS, testing on 5 testing datasets of DUTS, DUT-OMRON, HKU-IS, etc. 

Saillxl

the reasoning effect diagram mask is very fuzzy

1

Dear developers, the reasoning effect diagram mask is very fuzzy, which is caused by what! Looking forward to your reply, thank you

hczyni

Can this model resume from former checkpoint?

2

I don't see any function or parameters to load from existed checkpoint.

scpoppin1207

Questions about FEDER

2

Hi. In FEDER, there is a loss about the edge ( loss_edge = dice_loss(preds[6], edges)*0.125 + dice_loss(preds[7], edges)*0.25 + dice_loss(preds[8], edges)*0.5 ). However, not all datasets contain data about edge...

waner1998

wrong checkpoint

2

Thank you for your wonderful work. 'Encoder. SAM2-UNet applys the Hiera [38] backbone pretrained by SAM2.' However, I download the checkpoint from 'https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_large.pt' in 'https://github.com/facebookresearch/sam2/tree/main/checkpoints', but failed to init the...

waner1998

how to train with tiny checkpoints?

2

i have done the following changes: 1. https://github.com/WZH0120/SAM2-UNet/blob/eb1c38d870358cbdd769c9721062f7bb888ef9b5/train.py#L15 2. edit the yaml https://github.com/WZH0120/SAM2-UNet/blob/eb1c38d870358cbdd769c9721062f7bb888ef9b5/SAM2UNet.py#L127 but errors occur like python3.9/site-packages/torch/nn/modules/module.py", line 2215, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError:...

FeiYull

医学图像

7

感谢作者的开源代码：我运行了2分类的2D医学图像数据集，训练集大概两三百张，eval.py的结果很差：mDice: 0.168；mIoU: 0.111。请问有人知道原因吗？

Saillxl

Convert to ONNX and TensorRT

2

Hi, Is it possible to convert this model to ONNX or to TensorRT ? The architecture of this model seems a bit exotic and I'm having trouble converting it to...

pedrolsantos