From 5f2d3e890d897e854dfe30056a8765a5fbbea68c Mon Sep 17 00:00:00 2001 From: Askill Date: Sat, 19 Sep 2020 23:13:44 +0200 Subject: [PATCH] delete images after creating video --- generate test footage/gen.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/generate test footage/gen.py b/generate test footage/gen.py index f4e5a92..f7c4f87 100644 --- a/generate test footage/gen.py +++ b/generate test footage/gen.py @@ -18,6 +18,8 @@ length = .1 numberOfEvents = 3 dirname = os.path.dirname(__file__) +imageType = ".png" + imagesPath = os.path.join(dirname, 'images')+"/" outputPath = os.path.join(dirname, 'out.mp4') @@ -50,7 +52,7 @@ def genImages(): img1 = ImageDraw.Draw(img) img1.rectangle(objectShape, fill = color) - img.save( imagesPath + str(counter).zfill(6) + ".png") + img.save( imagesPath + str(counter).zfill(6) + imageType) def makeVideo(): fileList = [] @@ -64,5 +66,11 @@ def makeVideo(): writer.append_data(imageio.imread(im)) writer.close() +def deleteImages(): + filelist = glob.glob(os.path.join(imagesPath, "*" + imageType)) + for f in filelist: + os.remove(f) + genImages() makeVideo() +#deleteImages()