From e68d72305680e3ffff347d0e66eb9268ba52b9d0 Mon Sep 17 00:00:00 2001
From: Iridoudou <2534936416@qq.com>
Date: Sat, 7 Aug 2021 21:30:46 +0800
Subject: [PATCH] update code

---
 fit/configs/UTD_MHAD.json |  2 +-
 fit/tools/map.py          | 14 --------------
 fit/tools/save.py         | 12 ++++--------
 3 files changed, 5 insertions(+), 23 deletions(-)
 delete mode 100644 fit/tools/map.py

diff --git a/fit/configs/UTD_MHAD.json b/fit/configs/UTD_MHAD.json
index fc91d77..0e175fb 100644
--- a/fit/configs/UTD_MHAD.json
+++ b/fit/configs/UTD_MHAD.json
@@ -4,7 +4,7 @@
     },
     "TRAIN": {
         "LEARNING_RATE": 2e-2,
-        "MAX_EPOCH": 500,
+        "MAX_EPOCH": 10,
         "WRITE": 1
     },
     "USE_GPU": 1,
diff --git a/fit/tools/map.py b/fit/tools/map.py
deleted file mode 100644
index 15e998c..0000000
--- a/fit/tools/map.py
+++ /dev/null
@@ -1,14 +0,0 @@
-import numpy as np
-
-def mapping(Jtr,cfg):
-    name=cfg.DATASET.NAME
-    if not name=='HumanAct12':
-        mapped_joint=cfg.DATASET.DATA_MAP.UTD_MHAD
-        Jtr_mapped=np.zeros([Jtr.shape[0],len(mapped_joint),Jtr.shape[2]])
-        for i in range(Jtr.shape[0]):
-            for j in range(len(mapped_joint)):
-                for k in range(Jtr.shape[2]):
-                    Jtr_mapped[i][j][k]=Jtr[i][mapped_joint[j]][k]
-        return Jtr_mapped
-    return Jtr
-
diff --git a/fit/tools/save.py b/fit/tools/save.py
index 8d11ac8..471f791 100644
--- a/fit/tools/save.py
+++ b/fit/tools/save.py
@@ -41,14 +41,10 @@ def save_params(res, file, logger, dataset_name):
     fit_path = "fit/output/{}/params/".format(dataset_name)
     create_dir_not_exist(fit_path)
     logger.info('Saving params at {}'.format(fit_path))
-    pose_params = pose_params.cpu().detach()
-    pose_params = pose_params.numpy().tolist()
-    shape_params = shape_params.cpu().detach()
-    shape_params = shape_params.numpy().tolist()
-    Jtr = Jtr.cpu().detach()
-    Jtr = Jtr.numpy().tolist()
-    verts = verts.cpu().detach()
-    verts = verts.numpy().tolist()
+    pose_params = (pose_params.cpu().detach()).numpy().tolist()
+    shape_params = (shape_params.cpu().detach()).numpy().tolist()
+    Jtr = (Jtr.cpu().detach()).numpy().tolist()
+    verts = (verts.cpu().detach()).numpy().tolist()
     params = {}
     params["pose_params"] = pose_params
     params["shape_params"] = shape_params