updated

.gitignore

@@ -11,3 +11,5 @@ _ext
 tmp
 tmp
 *.o*
 *.o*
 *~
 *~
+.idea
+.DS_Store
\ No newline at end of file

.idea/misc.xml

 <?xml version="1.0" encoding="UTF-8"?>
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
 <project version="4">
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (pytorch)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (deepml)" project-jdk-type="Python SDK" />
   <component name="PyCharmProfessionalAdvertiser">
   <component name="PyCharmProfessionalAdvertiser">
     <option name="shown" value="true" />
     <option name="shown" value="true" />
   </component>
   </component>

.idea/pytorch_stacked_hourglass-master.iml

@@ -4,7 +4,7 @@
     <content url="file://$MODULE_DIR$">
     <content url="file://$MODULE_DIR$">
       <sourceFolder url="file://$MODULE_DIR$" isTestSource="false" />
       <sourceFolder url="file://$MODULE_DIR$" isTestSource="false" />
     </content>
     </content>
-    <orderEntry type="jdk" jdkName="Python 3.9 (pytorch)" jdkType="Python SDK" />
+    <orderEntry type="jdk" jdkName="Python 3.9 (deepml)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
   </component>
   <component name="PyDocumentationSettings">
   <component name="PyDocumentationSettings">

handler.py

 import cv2
 import cv2
-import utils.img
 import torch
 import torch
-import numpy as np
-from utils.group import HeatmapParser
+
 import data.MPII.ref as ds
 import data.MPII.ref as ds
+import utils.img
+from utils.group import HeatmapParser
 from utils.posture import *
 from utils.posture import *
 
 
 # 歪头的斜率阈值
 # 歪头的斜率阈值
@@ -11,12 +11,8 @@ CROOKED_HEAD_THRE=8
 # 斜肩的斜率阈值
 # 斜肩的斜率阈值
 OBLIQUE_SHOULDER_THRE=2
 OBLIQUE_SHOULDER_THRE=2
 
 
-
 # 人体骨骼各坐标点的下标对应图
 # 人体骨骼各坐标点的下标对应图
 
 
-
-
-
 class PersonPosture():
 class PersonPosture():
     def __init__(self, pred,orig_img):
     def __init__(self, pred,orig_img):
         self.pred = pred[0]['keypoints']
         self.pred = pred[0]['keypoints']
@@ -165,7 +161,6 @@ def main():
         return func(0, config, 'inference', imgs=torch.Tensor(np.float32(imgs)))['preds']
         return func(0, config, 'inference', imgs=torch.Tensor(np.float32(imgs)))['preds']
 
 
     def do(img, c, s):
     def do(img, c, s):
-        import time
         ans = inference(img, runner, config, c, s)
         ans = inference(img, runner, config, c, s)
         if len(ans) > 0:
         if len(ans) > 0:
             ans = ans[:,:,:3]
             ans = ans[:,:,:3]
@@ -176,7 +171,7 @@ def main():
             pred.append({'keypoints': ans[i,:,:]})
             pred.append({'keypoints': ans[i,:,:]})
         return pred
         return pred
 if __name__ == '__main__':
 if __name__ == '__main__':
-    image_path = "data/custom/wenjin正脸.jpg"
+    image_path = "data/custom/O型腿3.jpeg"
     from train import init
     from train import init
     func, config = init()
     func, config = init()