rename mobilenet

ZexinChen · ZexinChen · commit 351fc4c24441 · 2018-03-16T22:49:11.000+08:00
diff --git a/README.md b/README.md
@@ -1,5 +1,8 @@
-# DeepPose-pytorch
-Reproduce Google's [DeepPose](https://arxiv.org/pdf/1312.4659.pdf) with Resnet18 by Pytorch
+# MobilePose
+
+MobilePose is a **Tiny** PyTorch implementation of single person 2D pose estimation framework. The aim is to provide the interface of the training/inference/evaluation, and the dataloader with various data augmentation options. And final trained model can satisfy basic requirements(speed+size+accuracy) for mobile device.
+
+Some codes for mobilenetV2 and display are brought from [pytorch-mobilenet-v2](https://github.com/tonylins/pytorch-mobilenet-v2) and [tf-pose-estimation](https://github.com/ildoonet/tf-pose-estimation). Thanks to the original authors.
 
 ## Functionality
 
@@ -8,16 +11,20 @@ Reproduce Google's [DeepPose](https://arxiv.org/pdf/1312.4659.pdf) with Resnet18
 3. **Accurate** keypoint estimation (75~85mAP(0.5IoU))
 
 ## Requirements:
+
 - Python 3.6.2
 - Pytorch 0.2.0\_3 
+- imgaug 0.2.5
 
 ## Todo List:
-- [x] accuracy(mAP) evaluation script
-- [x] multi-scales training
-- [x] data augmentation(rotate/shift/flip/multi-scale)
-- [x] support LSP dataset
-- [x] adding weighted loss(coco keypoints weight) 
-- [x] support Macbook camera realtime skeleton display demo
+
+- [x] multi-thread dataloader
+- [x] training and inference
+- [x] performance evaluation
+- [x] multi-scale training
+- [x] support resnet18/mobilenetV2
+- [x] data augmentation(rotate/shift/flip/multi-scale/noise)
+- [x] Macbook camera realtime display script
 
 ## Usage
 
@@ -27,10 +34,18 @@ export CUDA_VISIBLE_DEVICES=0; python training.py --model=mobilenet/resnet --gpu
 ```
 2. Evaluation
 ```shell
-python eval.py --model=mobilenet/resnet
+export CUDA_VISIBLE_DEVICES=0; python eval.py --model=mobilenet/resnet
 ```
 4. Realtime visualization:
 ```shell
 python run_webcam.py --model=mobilenet/resnet
 ```
 
+## Contributors
+
+MobilePose is developed and maintained by [Yuliang Xiu](http://xiuyuliang.cn/about/), [Zexin Chen](https://github.com/ZexinChen) and [Yinghong Fang](https://github.com/Fangyh09).
+
+## License
+
+MobilePose is freely available for free non-commercial use. For commercial queries, please contact [Cewu Lu](http://www.mvig.org/) or [SightPlus Co. Ltd](https://www.sightp.com/).
+
diff --git a/coco_utils.py b/coco_utils.py
@@ -1,6 +1,6 @@
 '''
 File: coco_utils.py
-Project: DeepPose
+Project: MobilePose
 File Created: Saturday, 3rd March 2018 7:04:57 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
diff --git a/dataloader.py b/dataloader.py
@@ -1,6 +1,6 @@
 '''
 File: dataloader.py
-Project: DeepPose
+Project: MobilePose
 File Created: Thursday, 8th March 2018 3:00:27 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
diff --git a/estimator.py b/estimator.py
@@ -1,6 +1,6 @@
 '''
 File: estimator.py
-Project: DeepPose
+Project: MobilePose
 File Created: Thursday, 8th March 2018 3:02:01 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
diff --git a/eval.py b/eval.py
@@ -1,7 +1,7 @@
 # coding: utf-8
 '''
 File: eval.py
-Project: DeepPose
+Project: MobilePose
 File Created: Thursday, 8th March 2018 1:54:07 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
@@ -42,7 +42,7 @@
 
 if __name__ == '__main__':
     
-    parser = argparse.ArgumentParser(description='deeppose demo')
+    parser = argparse.ArgumentParser(description='MobilePose Demo')
     parser.add_argument('--model', type=str, default="resnet")
     args = parser.parse_args()
     modeltype = args.model
@@ -53,9 +53,9 @@
 
     PATH_PREFIX = "./results/{}".format(modeltype)
     full_name="./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/mobilenet-best.t7"
+    # full_name = "/home/yuliang/code/MobilePose-pytorch/models/demo/mobilenetv2_224x224-robust.t7"
+    # full_name = "/home/yuliang/code/MobilePose-pytorch/models/demo/resnet18_227x227.t7"
+    # full_name = "/home/yuliang/code/MobilePose-pytorch/models/demo/mobilenet-best.t7"
     # ROOT_DIR = "/home/yuliang/code/deeppose_tf/datasets/mpii"
     ROOT_DIR = "../deeppose_tf/datasets/mpii"
     
diff --git a/mobilenetv2.py b/mobilenetv2.py
@@ -1,6 +1,6 @@
 '''
 File: mobilenetv2.py
-Project: DeepPose
+Project: MobilePose
 File Created: Thursday, 8th March 2018 2:51:18 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
diff --git a/networks.py b/networks.py
@@ -1,6 +1,6 @@
 '''
 File: networks.py
-Project: DeepPose
+Project: MobilePose
 File Created: Thursday, 8th March 2018 2:59:28 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
diff --git a/run_webcam.py b/run_webcam.py
@@ -1,6 +1,6 @@
 '''
 File: run_webcam.py
-Project: DeepPose
+Project: MobilePose
 File Created: Thursday, 8th March 2018 2:19:39 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
@@ -28,7 +28,7 @@
 
 if __name__ == '__main__':
     
-    parser = argparse.ArgumentParser(description='deeppose realtime webcam.')
+    parser = argparse.ArgumentParser(description='MobilePose Realtime Webcam.')
     parser.add_argument('--model', type=str, default='resnet', help='mobilenet|resnet')
     parser.add_argument('--camera', type=int, default=0)
 
diff --git a/training.py b/training.py
@@ -2,7 +2,7 @@
 
 '''
 File: training.py
-Project: DeepPose
+Project: MobilePose
 File Created: Thursday, 8th March 2018 2:50:11 pm
 Author: Yuliang Xiu (yuliangxiu@sjtu.edu.cn)
 -----
@@ -29,7 +29,7 @@
 
 if __name__ == '__main__':
 
-    parser = argparse.ArgumentParser(description='deeppose demo')
+    parser = argparse.ArgumentParser(description='MobilePose Demo')
     parser.add_argument('--model', type=str, default="resnet")
     parser.add_argument('--gpu', type=str, default="0")
     args = parser.parse_args()
