YXu556 / SITS-MoCo

23 stars 2 forks source link

调试代码的问题? #2

Open zhangyang5858 opened 5 months ago

zhangyang5858 commented 5 months ago

近期拜读到您写在ISPRS上Self-supervised pre-training for large-scale crop mapping using Sentinel-2 time series文章,想要尝试自己使用代码复现一下,但是在使用代码里面的main_moco.py代码时,使用get_moco_model函数建立MoCo数据集时程序会报错,具体是在builder.py里面的self.encoder_q = base_encoder()这一行,显示base_encoder()函数找不到而导致模型报错,请问我要进行哪方面的修改从而能正常使用给的模型代码?

YXu556 commented 5 months ago

您好, get_moco_model函数用于建立MoCo训练模型。您可以尝试以下修改: 1) 如果您使用提供的模型,请确保在运行时指定正确的模型名称,如python main_moco.py transformer --rc --use-doy --useall --mlp指定--model 参数为transformer。 2) 如果您尝试使用自己的模型,需要在get_moco_model函数中的条件语句中添加对该模型名称的识别。

zhangyang5858 commented 5 months ago

你好: 谢谢修改上传的下载S2影像的代码,但是复现数据集过程中还有一个问题,在preprocess文件里面0. blockpartition.py代码里面有一行是Path(r"..\data\US\Boundary"),请问这里美国边界数据要在哪里下载?