From 5847dc57719cf74944fef34a17f5d47c5956442a Mon Sep 17 00:00:00 2001 From: zjut Date: Fri, 15 Nov 2024 17:43:21 +0800 Subject: [PATCH] =?UTF-8?q?test(sar):=20=E6=9B=B4=E6=96=B0=E6=A8=A1?= =?UTF-8?q?=E5=9E=8B=E5=B9=B6=E6=B5=8B=E8=AF=95sar=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=E9=9B=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 更新模型权重文件路径 - 修改测试数据集为仅使用sar数据集 - 更改模型名称描述 --- net.py | 1 + test_IVF.py | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/net.py b/net.py index 09bee70..13ab4ab 100644 --- a/net.py +++ b/net.py @@ -426,6 +426,7 @@ class Restormer_Encoder(nn.Module): self.baseFeature_sar = BaseFeatureExtractionSAR(dim=dim) self.detailFeature_sar = DetailFeatureExtractionSAR() + def forward(self, inp_img,is_sar=False): inp_enc_level1 = self.patch_embed(inp_img) out_enc_level1 = self.encoder_level1(inp_enc_level1) diff --git a/test_IVF.py b/test_IVF.py index c402ae7..99c91bd 100644 --- a/test_IVF.py +++ b/test_IVF.py @@ -17,11 +17,11 @@ current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") os.environ["CUDA_VISIBLE_DEVICES"] = "0" -ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion10-08-16-20.pth" +ckpt_path= r"/home/star/whaiDir/PFCFuse/models/whaiFusion11-15-09-38.pth" -for dataset_name in ["TNO","RoadScene"]: +for dataset_name in ["sar"]: print("\n"*2+"="*80) - model_name="PFCFuse " + model_name="base 基础，没有修改 " print("The test result of "+dataset_name+' :') test_folder = os.path.join('test_img', dataset_name) test_out_folder=os.path.join('test_result',current_time,dataset_name)