From ad631f41e183ee84a5efd9d5a3d8aa0a9effb406 Mon Sep 17 00:00:00 2001
From: lzzyzlbb <287246233@qq.com>
Date: Wed, 8 Dec 2021 19:59:06 +0800
Subject: [PATCH] fix tipc (#517)

* fix tipc

* add cyclegan eval
---
 configs/cyclegan_horse2zebra.yaml               | 11 ++++++++---
 ppgan/models/cycle_gan_model.py                 | 10 ++++++++++
 test_tipc/configs/msvsr/train_infer_python.txt  |  8 ++++----
 test_tipc/results/python_msvsr_results_fp32.txt |  4 ++--
 4 files changed, 24 insertions(+), 9 deletions(-)

diff --git a/configs/cyclegan_horse2zebra.yaml b/configs/cyclegan_horse2zebra.yaml
index f5c8054..e0b0c29 100644
--- a/configs/cyclegan_horse2zebra.yaml
+++ b/configs/cyclegan_horse2zebra.yaml
@@ -120,6 +120,11 @@ log_config:
 snapshot_config:
   interval: 5
 
-export_model:
-  - {name: 'netG_A', inputs_num: 1}
-  - {name: 'netG_B', inputs_num: 1}
+validate:
+  interval: 30000
+  save_img: false
+  metrics:
+    fid: # metric name, can be arbitrary
+        name: FID
+        batch_size: 8
+
diff --git a/ppgan/models/cycle_gan_model.py b/ppgan/models/cycle_gan_model.py
index 6d1c3f0..000938e 100644
--- a/ppgan/models/cycle_gan_model.py
+++ b/ppgan/models/cycle_gan_model.py
@@ -242,3 +242,13 @@ class CycleGANModel(BaseModel):
         self.backward_D_B()
         # update D_A and D_B's weights
         optimizers['optimD'].step()
+
+
+    def test_iter(self, metrics=None):
+        self.nets['netG_A'].eval()
+        self.forward()
+        with paddle.no_grad():
+            if metrics is not None:
+                for metric in metrics.values():
+                    metric.update(self.fake_B, self.real_B)
+        self.nets['netG_A'].train()
diff --git a/test_tipc/configs/msvsr/train_infer_python.txt b/test_tipc/configs/msvsr/train_infer_python.txt
index a848037..0f393ab 100644
--- a/test_tipc/configs/msvsr/train_infer_python.txt
+++ b/test_tipc/configs/msvsr/train_infer_python.txt
@@ -13,7 +13,7 @@ train_infer_img_dir:./data/msvsr_reds/test
 null:null
 ##
 trainer:norm_train
-norm_train:tools/main.py -c configs/msvsr_reds.yaml --seed 123 -o dataset.train.dataset.num_clips=2 dataset.train.num_workers=0 log_config.interval=1 snapshot_config.interval=5 
+norm_train:tools/main.py -c configs/msvsr_reds.yaml --seed 123 -o dataset.train.dataset.num_clips=2 dataset.train.num_workers=0 log_config.interval=1 snapshot_config.interval=5 dataset.train.dataset.number_frames=2 
 pact_train:null
 fpgm_train:null
 distill_train:null
@@ -27,7 +27,7 @@ null:null
 ===========================infer_params===========================
 --output_dir:./output/
 load:null
-norm_export:tools/export_model.py -c configs/msvsr_reds.yaml  --inputs_size="1,4,3,180,320" --load 
+norm_export:tools/export_model.py -c configs/msvsr_reds.yaml  --inputs_size="1,2,3,180,320" --load 
 quant_export:null 
 fpgm_export:null
 distill_export:null
@@ -37,7 +37,7 @@ inference_dir:multistagevsrmodel_generator
 train_model:./inference/msvsr/multistagevsrmodel_generator
 infer_export:null
 infer_quant:False
-inference:tools/inference.py --model_type msvsr -c configs/msvsr_reds.yaml --seed 123 -o dataset.test.num_clips=2 dataset.test.number_frames=4 --output_path test_tipc/output/
+inference:tools/inference.py --model_type msvsr -c configs/msvsr_reds.yaml --seed 123 -o dataset.test.num_clips=2 dataset.test.number_frames=2 --output_path test_tipc/output/
 --device:gpu
 null:null
 null:null
@@ -48,4 +48,4 @@ null:null
 null:null
 null:null
 --benchmark:True
-null:null
\ No newline at end of file
+null:null
diff --git a/test_tipc/results/python_msvsr_results_fp32.txt b/test_tipc/results/python_msvsr_results_fp32.txt
index 0de2d2f..1462d70 100644
--- a/test_tipc/results/python_msvsr_results_fp32.txt
+++ b/test_tipc/results/python_msvsr_results_fp32.txt
@@ -1,2 +1,2 @@
-Metric psnr: 27.3670
-Metric ssim: 0.8021
\ No newline at end of file
+Metric psnr: 24.3250
+Metric ssim: 0.6497
\ No newline at end of file
-- 
GitLab