Revert "fix PP-HumanV2 PP-TinyPose PP-Vehicle video adaptive code (#5689)"

This reverts commit 7219a0e1.

modelcenter/PP-HumanV2/APP/pipeline/pipeline.py

@@ -389,7 +389,8 @@ class PipePredictor(object):
 
     def set_file_name(self, path):
         if path is not None:
-            self.file_name = path
+            if "." in path:
+                self.file_name = path.split(".")[-2]
         else:
             # use camera id
             self.file_name = None
@@ -445,9 +446,9 @@ class PipePredictor(object):
         frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
         print("video fps: %d, frame_count: %d" % (fps, frame_count))
 
-        video_name,suffix = os.path.splitext(self.file_name)
-        out_path = video_name + "_output"+suffix
-        fourcc = int(capture.get(cv2.CAP_PROP_FOURCC))
+        video_out_name = 'output' if self.file_name is None else self.file_name
+        out_path = video_out_name + "_output.mp4"
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         writer = cv2.VideoWriter(out_path, fourcc, fps, (width, height))
 
         frame_id = 0

modelcenter/PP-TinyPose/APP/det_keypoint_unite_infer.py

@@ -107,18 +107,18 @@ def topdown_unite_predict_video(FLAGS,
         capture = cv2.VideoCapture(camera_id)
     else:
         capture = cv2.VideoCapture(FLAGS.video_file)
-        video_name,suffix = os.path.splitext(FLAGS.video_file) 
-        video_name = video_name+"_output"+suffix
-    
+        video_name,suffix = FLAGS.video_file.split('.')
+        video_name = video_name+"_output."+suffix
     # Get Video info : resolution, fps, frame count
     width = int(capture.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(capture.get(cv2.CAP_PROP_FRAME_HEIGHT))
     fps = int(capture.get(cv2.CAP_PROP_FPS))
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
     print("fps: %d, frame_count: %d" % (fps, frame_count))
-    
-    codec = int(capture.get(cv2.CAP_PROP_FOURCC))
-    writer = cv2.VideoWriter(video_name, codec, fps, (width, height))
+
+   
+    fourcc = cv2.VideoWriter_fourcc(* 'mp4v')
+    writer = cv2.VideoWriter(video_name, fourcc, fps, (width, height))
     index = 0
     store_res = []
     keypoint_smoothing = KeypointSmoothing(

modelcenter/PP-Vehicle/APP/pipeline/pipeline.py

@@ -329,7 +329,8 @@ class PipePredictor(object):
 
     def set_file_name(self, path):
         if path is not None:
-            self.file_name = path
+            if "." in path:
+                self.file_name = path.split(".")[-2]
         else:
             # use camera id
             self.file_name = None
@@ -396,9 +397,9 @@ class PipePredictor(object):
         frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
         print("video fps: %d, frame_count: %d" % (fps, frame_count))
 
-        video_name,suffix = os.path.splitext(self.file_name)
-        out_path = video_name + "_output"+suffix
-        fourcc = int(capture.get(cv2.CAP_PROP_FOURCC))
+        video_out_name = 'output' if self.file_name is None else self.file_name
+        out_path = video_out_name + "_output.mp4"
+        fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         writer = cv2.VideoWriter(out_path, fourcc, fps, (width, height))
 
         frame_id = 0