add comment

Author: YuliangXiu

eval.py

@@ -10,6 +10,7 @@
 -----
 Copyright 2018 - 2018 Shanghai Jiao Tong University, Machine Vision and Intelligence Group
 '''
+
 import warnings
 warnings.filterwarnings('ignore')
 
@@ -53,7 +54,8 @@
     PATH_PREFIX = "/home/yuliang/code/DeepPose-pytorch/results/{}".format(modeltype)
     full_name="/home/yuliang/code/DeepPose-pytorch/models/{}/{}".format(modeltype, filename)
     # full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/mobilenetv2_224x224-robust.t7"
-    full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/resnet18_227x227.t7"
+    # full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/resnet18_227x227.t7"
+    full_name = "/home/yuliang/code/DeepPose-pytorch/models/demo/mobilenet-best.t7"
     ROOT_DIR = "/home/yuliang/code/deeppose_tf/datasets/mpii"
 
     if modeltype == 'resnet':
@@ -66,8 +68,8 @@
     # load dataset
     test_dataset = PoseDataset(csv_file=os.path.join(ROOT_DIR,'test_joints.csv'),
                                 transform=transforms.Compose([
-                                            Rescale((input_size, input_size)), # for resnet18
-                                            # Wrap((input_size,input_size)), # for mobilenet
+                                            # Rescale((input_size, input_size)), # for resnet18 and mobilenet
+                                            Wrap((input_size,input_size)), # only for mobilenet-best
                                             Expansion(),
                                             ToTensor()
                                         ]))

run_webcam.py

@@ -42,16 +42,10 @@
     image = crop_camera(image)
 
     while True:
-        fps_time = time.time()
         ret_val , image = cam.read()
         image = crop_camera(image)
         humans = e.inference(image, args.model)
         image = ResEstimator.draw_humans(image, humans, imgcopy=False)
-
-        cv2.putText(image,
-                    "FPS: %f" % (1.0 / (time.time() - fps_time)),
-                    (10, 10),  cv2.FONT_HERSHEY_SIMPLEX, 0.5,
-                    (0, 255, 0), 2)
         cv2.imshow('tf-pose-estimation result', image)
         if cv2.waitKey(1) == 27: # ESC
             break