reorganize the code

dorianhenning · Jul 11, 2018 · 917ad4b · 917ad4b
1 parent 33ef1b2
commit 917ad4b
Show file tree

Hide file tree

Showing 91 changed files with 19 additions and 177,578 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,5 +1,6 @@
 human_detection/output
 examples/results
+examples/res
 PoseFlow/__pycache__
 PoseFlow/*.npy
 PoseFlow/alpha-pose-results-test.json

diff --git a/README.md b/README.md
@@ -28,16 +28,16 @@ To match poses that correspond to the same person across frames, we also provide
   ./install.sh
   ```
 
-1. Download the models manually: **ssd_coco.pth**([Google Drive](https://drive.google.com/open?id=1ifUNnxSDOP7InBtKy5CGnRAMLD4-d4Ct) | [Baidu pan](https://pan.baidu.com/s/1mon7Ht6ObqrS2ZY955swpg)), **pyra_4.pth** ([Google Drive](https://drive.google.com/open?id=1oG1Fxj4oBfKwD1W_2QObxltWybuIk7Y6) | [Baidu pan](https://pan.baidu.com/s/14ONL_T_d1twm9Lxac5x-Ew)), **yolov3.weights**([Google Drive](https://drive.google.com/open?id=1yjrziA2RzFqWAQG4Qq7XN0vumsMxwSjS) | [Baidu pan](https://pan.baidu.com/s/108SjV-uIJpxnqDMT19v-Aw)). Place them into `./models/ssd`, `./models/sppe` and `./models/yolo` respectively.
+1. Download the models manually: **pyra_4.pth** ([Google Drive](https://drive.google.com/open?id=1oG1Fxj4oBfKwD1W_2QObxltWybuIk7Y6) | [Baidu pan](https://pan.baidu.com/s/14ONL_T_d1twm9Lxac5x-Ew)), **yolov3.weights**([Google Drive](https://drive.google.com/open?id=1yjrziA2RzFqWAQG4Qq7XN0vumsMxwSjS) | [Baidu pan](https://pan.baidu.com/s/108SjV-uIJpxnqDMT19v-Aw)). Place them into `./models/sppe` and `./models/yolo` respectively.
 
 
 ## Quick Start
 - **Demo**:  Run AlphaPose for all images in a folder and visualize the results with:
 ```
-python demo_yolo.py \
---inputlist ./list-coco-minival500.txt \
+python3 demo.py \
+--inputlist examples/list-coco-demo.txt \
 --imgpath ${img_directory} \
---outputpath ./coco-minival
+--outputpath examples/res
 ```
 
 

diff --git a/demo_yolo.py → demo.py b/demo_yolo.py → demo.py
diff --git a/demo_fast.py b/demo_fast.py
diff --git a/demo_faster.py b/demo_faster.py
diff --git a/examples/img-list.txt b/examples/img-list.txt
diff --git a/list-coco-demo.txt → examples/list-coco-demo.txt b/list-coco-demo.txt → examples/list-coco-demo.txt
diff --git a/list-coco-minival500.txt → examples/list-coco-minival500.txt b/list-coco-minival500.txt → examples/list-coco-minival500.txt
diff --git a/fetch_models.sh b/fetch_models.sh
diff --git a/fn.py b/fn.py
@@ -110,15 +110,15 @@ def vis_res(final_result, outputpath, format='coco'):
                     continue
                 cor_x, cor_y = int(kp_preds[n, 0]), int(kp_preds[n, 1])
                 part_line[n] = (cor_x, cor_y)
-                cv2.circle(img, (cor_x, cor_y), 6, p_color[n], -1)
+                cv2.circle(img, (cor_x, cor_y), 4, p_color[n], -1)
             # Draw limbs
             for start_p, end_p in l_pair:
                 if start_p in part_line and end_p in part_line:
                     start_p = part_line[start_p]
                     end_p = part_line[end_p]
                     cv2.line(img, start_p, end_p, YELLOW, 2)
 
-        cv2.imwrite(os.path.join(outputpath, 'res' + im_name), img)
+        cv2.imwrite(os.path.join(outputpath, im_name), img)
 
 
 def getTime(time1=0):

diff --git a/install.sh b/install.sh
@@ -1,5 +1,5 @@
-pip install torch
-pip install torchvision
-pip install -e git+https://github.com/ncullen93/torchsample.git#egg=torchsample
-pip install visdom
-pip install nibabel
+pip3 install --user torch
+pip3 install --user torchvision
+pip3 install --user -e git+https://github.com/ncullen93/torchsample.git#egg=torchsample
+pip3 install --user visdom
+pip3 install --user nibabel