search

ViTAE-Transformer / SAMRS

The official repo for [NeurIPS'23] "SAMRS: Scaling-up Remote Sensing Segmentation Dataset with Segment Anything Model"
288 stars 14 forks source link

ZeroDivisionError: float division by zero #36

Open 1835969208 opened 3 months ago

1835969208 commented 3 months ago

作者大人，我在运行main_finetune.py时候遇到报错。具体是main_finetune.py表明在计算 F1 分数时出现了除零错误。我怀疑是我postdam数据集的结构有问题吗？下图是我的文件夹结构。其中images文件夹下保存rbg图像（格式jpg），train.txt和valid.txt分别保存训练集和验证集图片的名称，lables文件夹下保存images文件夹中相同图像的分割图像。 image

1835969208 commented 3 months ago

同时想问一下，main_finetune.py文件中170行和215的root目录应该如何设置。是postdam这个根目录还是下面的images文件夹的地址

DotWang commented 3 months ago

@1835969208 检查标签问题：https://github.com/ViTAE-Transformer/ViTAE-Transformer-Remote-Sensing/issues/21

DotWang commented 3 months ago

@1835969208 finetune可以不需要train和val.txt，在image和label文件夹下边分别分开train和val就行

1835969208 commented 3 months ago

作者大人，请问有处理好的postdam数据集吗，就是可以直接上传使用的。请问可以给个链接吗

DotWang commented 3 months ago

@1835969208 这个真没有，这是公开数据集，都是自己写脚本处理

1835969208 commented 3 months ago

作者大人，想问一下你是怎么处理postdam数据集的，因为readme里面没有写，其他issue里面描述的也比较杂乱。或者可以上传一下您用脚本处理好的postdam数据集吗。十分感谢

DotWang commented 3 months ago

@1835969208 这个issue里已经写了我们的做法和mmseg的做法的区别，https://github.com/ViTAE-Transformer/ViTAE-Transformer-Remote-Sensing/issues/21/，数据集是不可能上传的，超出了本项目的范畴，其他人也没有这方面的问题，都是自己写脚本处理

1835969208 commented 3 months ago

好的，感谢博主