update model

NTED/NTED_module.py

@@ -43,8 +43,8 @@ class NTED():
         
         self.data_loader = DemoDataset()
         
-        mp_hands = mp.solutions.hands
-        self.hands = mp_hands.Hands(static_image_mode=True, max_num_hands=2, min_detection_confidence=0.1) 
+        # mp_hands = mp.solutions.hands
+        # self.hands = mp_hands.Hands(static_image_mode=True, max_num_hands=2, min_detection_confidence=0.1) 
         
         self.ref_img = cv2.imread('example/ref_img.png')
         self.ref_img = cv2.resize(self.ref_img, (352, 512))
@@ -84,9 +84,9 @@ class NTED():
         
         body_pose, bbox = self.openpose_module.detect_body_pose(img.copy())
         
-        hand_pose = self.hand_pose_est(img.copy())
+        # hand_pose = self.hand_pose_est(img.copy())
         
-        data = self.data_loader.load_item(self.ref_img, body_pose[0], hand_pose)
+        data = self.data_loader.load_item(self.ref_img, body_pose[0], None)
 
         output = self.net_G(
             data['reference_image'], 

NTED/fashion_512.yaml

@@ -70,7 +70,7 @@ gen:
     type: NTED.extraction_distribution_model::Generator
     param:
       size: 512
-      semantic_dim: 30
+      semantic_dim: 20
       channels:
         16: 512
         32: 512
@@ -124,6 +124,6 @@ data:
     val:
       batch_size: 4
       distributed: True
-    hand_keypoint: True
+    hand_keypoint: False
 
 

NTED/nted_checkpoint.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:359d3d3bac365afe04aa8b906f1dc8891f0dd87ff1dfe5e60059b4fb9bb96af8
-size 284375285
+oid sha256:d7aad6b43e6f0187aac3696dbc4d2a531fef7ab3fe5824d1d1fff9ed7e97245a
+size 170512367

app.py

@@ -18,7 +18,7 @@ gr.Interface(fn=pose_transfer,
              title="谷小雨姿态驱动图像",
              examples=[["example/exp1.png"], ["example/exp2.png"], ["example/exp3.png"],\
                  ["example/exp4.png"], ["example/exp5.png"], ["example/exp6.png"]],
-             ).launch(server_name='0.0.0.0')  
+             ).launch(share=True)  # (server_name='0.0.0.0')  
 
 '''
 TODO