hitachinsk / SAMed

The implementation of the technical report: "Customized Segment Anything Model for Medical Image Segmentation"
MIT License
468 stars 43 forks source link

if elif sentence logic question #53

Open PreWisdom opened 2 months ago

PreWisdom commented 2 months ago

I find a question in SAMed/datasets/dataset_synapse.py line 38~40: if random.random() > 0.5: image, label = random_rot_flip(image, label) elif random.random() > 0.5: image, label = random_rotate(image, label) is that a logic problem or my misunderstanding?

PreWisdom commented 2 months ago

另外我还想请教一个问题,如果我做的是二分类分割,num_classes应该设置成1还是2?无论设置成1还是2都没有效果,预测出来的结果全是背景,没有目标前景

Saillxl commented 1 month ago

把学习率调到0.001试试