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