takaniwa / DSNet

a novel real-time model in semantic segmentation
MIT License
99 stars 3 forks source link

bdd10k #12

Open Liufanfanfan opened 1 month ago

Liufanfanfan commented 1 month ago

发现“train”等少数类别 在训练集和验证集中占比很少 导致最后该类别iou很小或者为0。请问怎么解决?

takaniwa commented 1 month ago

发现“train”等少数类别 在训练集和验证集中占比很少 导致最后该类别iou很小或者为0。请问怎么解决?

bdd10k数据集是有这个问题,具体你可以参考我的训练log。想解决这个问题,可以尝试使用类均匀采样训练。考虑到跟之前的工作公平对比,我没有实现这个代码。或许你可以尝试实现类均匀采样来解决这个问题。

Liufanfanfan commented 1 month ago

发现“train”等少数类别 在训练集和验证集中占比很少 导致最后该类别iou很小或者为0。请问怎么解决?

bdd10k数据集是有这个问题,具体你可以参考我的训练log。想解决这个问题,可以尝试使用类均匀采样训练。考虑到跟之前的工作公平对比,我没有实现这个代码。或许你可以尝试实现类均匀采样来解决这个问题。

感谢解答,请问可以上传camvid的log吗?我下载camvid数据集数据标签是0-11,包括背景类一共是12类,怎么只计算前11类的iou呢?

takaniwa commented 1 month ago

发现“train”等少数类别 在训练集和验证集中占比很少 导致最后该类别iou很小或者为0。请问怎么解决?

bdd10k数据集是有这个问题,具体你可以参考我的训练log。想解决这个问题,可以尝试使用类均匀采样训练。考虑到跟之前的工作公平对比,我没有实现这个代码。或许你可以尝试实现类均匀采样来解决这个问题。

感谢解答,请问可以上传camvid的log吗?我下载camvid数据集数据标签是0-11,包括背景类一共是12类,怎么只计算前11类的iou呢?

Camvid的log现在已经没了,比较久远了