In build_dataset.py (all_frames error)

[chahatagarwal]

In capture_process_frames function, it will not be able to manage more than 29 frames, hence have a condition check or push it under try block! Check the Code below

while vidObj.isOpened(): success, image = vidObj.read() if success : image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) image = resize_frame(image, size_frame-180, offset=35) image = cv2.resize(image, dsize=(size, size), interpolation=cv2.INTER_LINEAR) try: all_frames[countsize:size(count+1), :] = image except: break count += 1

[chahatagarwal]

In capture_process_frames function, it will not be able to manage more than 29 frames, hence have a condition check or push it under try block! Check the Code below

while vidObj.isOpened(): success, image = vidObj.read() if success : image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) image = resize_frame(image, size_frame-180, offset=35) image = cv2.resize(image, dsize=(size, size), interpolation=cv2.INTER_LINEAR) try: all_frames[count_size:size_(count+1), :] = image except: break count += 1

You can execute this script using below command: python build_dataset.py --data_dir data/ --output_dir output