summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--AutoVideosGPU.py7
1 files changed, 5 insertions, 2 deletions
diff --git a/AutoVideosGPU.py b/AutoVideosGPU.py
index 25ccabd..a088e65 100644
--- a/AutoVideosGPU.py
+++ b/AutoVideosGPU.py
@@ -92,8 +92,11 @@ def createVideo(width, height, videoLength, frameRate, functionLength,
clProgramTemplate2 = replace(clProgramTemplate2, '<RFUNCTION>', rfunction)
clProgramTemplate2 = replace(clProgramTemplate2, '<GFUNCTION>', gfunction)
clProgramTemplate2 = replace(clProgramTemplate2, '<BFUNCTION>', bfunction)
-
- fourcc = cv2.cv.CV_FOURCC(*'XVID')
+
+ try:
+ fourcc = cv2.cv.CV_FOURCC(*'XVID')
+ except:
+ fourcc = cv2.VideoWriter_fourcc(*'XVID')
name = ' '.join([namemaker3.generate() for i in range(3)]) + '.avi'
out = cv2.VideoWriter(directory+'/'+name,fourcc, frameRate, (width,height))