search

ViTAE-Transformer / SAMRS

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

AttributeError: 'SemsegPretrnFramework' object has no attribute 'decoder' #31

Open 1835969208 opened 1 month ago

1835969208 commented 1 month ago

作者大人，我在预训练时候根据您github上的示例命令选择--decoder 'upernet'。但报错显示SemsegPretrnFramework 类中没有对‘upernet’ decoder 的处理逻辑，我翻看了model.py文件也没有找到。所以想问一下作者大人。以下是我自己的运行命令：CUDA_VISIBLE_DEVICES=0 python main_pretrain.py --backbone 'resnet50' --decoder 'upernet' --datasets 'sota' 'sior' 'fast' --batch_size 12 --batch_size_val 12 --workers 8 --save_path '/root/autodl-tmp/SAMRs/save' --distributed 'False' --end_iter 80000 --image_size 224 --init_backbone 'imp' 屏幕截图 2024-08-03 180430

DotWang commented 1 month ago

@1835969208 main_pretrain.py不需要指定upernet，Readme写的很清楚

1835969208 commented 1 month ago

作者大人，readme里面好像是有指定的 ![Uploading 屏幕截图 2024-08-03 190240.png…]()

1835969208 commented 1 month ago

屏幕截图 2024-08-03 190240

1835969208 commented 1 month ago

不指定的话好像会运行不了 image

DotWang commented 1 month ago

@1835969208 我懂了，你搞错文件夹了，upernet应该用encoder_decoder那个文件夹

1722683847105
1835969208 commented 1 month ago

哦哦，万分感谢博主大人，对我这研0傻子太好了。另外还想问一下这encoder_decoder和end_to_end文件夹有什么区别吗

DotWang commented 1 month ago

@1835969208 encoder_decoder是传统卷积分割网络，unet，upernet这种的，end-to-end是纯transformer的，mask2former这种，在论文里边大部分的实验在encoder_decoder上完成