Add pose models and fix missing files bug

.gitignore

@@ -1,3 +1,2 @@
 *__pycache__*
 */model/
-*/models/

VITON_Demo_post.py

@@ -117,6 +117,7 @@ def run(self):
         while self.run_flag:
             count += 1
             ret, frame = self.cap.read()
+            frame = cv2.flip(frame, 0)
             if VIDEO_SOURCE in [0, 1]:
                 frame = np.rot90(frame, 3)
             else:
@@ -313,7 +314,7 @@ def __init__(self, pose_and_seg_data_queue,
         self.viton_frame_queue2 = viton_frame_queue2
         self.run_flag = True
         self.pause_flag = False
-        self.prod_name = './test_product.jpg'
+        self.prod_name = './inputs/a.jpg'
         if RECORD_VIDEO and False:
             current_time = strftime("%Y%m%d_%H%M", gmtime())
             output_dir = './inputs'

tf_pose_estimation/models/graph/cmu/download.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+
+echo "download model graph : cmu"
+
+extract_download_url() {
+
+        url=$( wget -q -O - $1 |  grep -o 'http*://download[^"]*' | tail -n 1 )
+        echo "$url"
+
+}
+
+wget $( extract_download_url http://www.mediafire.com/file/1pyjsjl0p93x27c/graph_freeze.pb ) -O graph_freeze.pb
+wget $( extract_download_url http://www.mediafire.com/file/qlzzr20mpocnpa3/graph_opt.pb ) -O graph_opt.pb
+wget $( extract_download_url http://www.mediafire.com/file/i72ll9k5i7x6qfh/graph.pb ) -O graph.pb