Hello I am trying to convert avi files into separate png files. My code is here:
for i in tqdm(range(num_videos)):
if not os.path.isdir(args.VideoDir):
print("Input folder does not exist!")
break
if not os.path.isdir(args.OutputDir):
os.mkdir(args.OutputDir)
path = args.VideoDir + videos[i]
path = args.VideoDir + videos[i]
images = io.read_video(path)[0]
if images.shape[0] < args.numframes:
for n in range(images.shape[0]):
folder_path = args.OutputDir + "scene_" + str(1000 + i)
if not os.path.isdir(folder_path):
os.mkdir(folder_path)
image = images[n]
image = torch.reshape(image, (image.shape[2], image.shape[0], image.shape[1]))
image_path = folder_path + "/" + "col_high_" + str(n).zfill(4) + ".png"
utils.save_image(image.float(), fp=image_path)
else:
for n in range(args.numframes):
folder_path = args.OutputDir + "scene_" + str(1000 + i)
if not os.path.isdir(folder_path):
os.mkdir(folder_path)
image = images[n]
image = image.view(image.shape[2], image.shape[0], image.shape[1])
image_path = folder_path + "/" + "col_high_" + str(n).zfill(4) + '.png'
utils.save_image(image.type(torch.float32), image_path)
if not args.keep_video:
os.remove(path)
The output image ends up being totally wrong. I have tried to do the same with mp4 files but the result doesn’t change. Here is an example image:
Any help would be appreciated. Thanks!
