[ Feature Suggestion ] Adding tqdm progress bar for more accurate completion time

[JD-2006]

Adding a progress bar that is more accurate and informative than the gradio one.

from tqdm import tqdm  # Import tqdm


# Process each frame with tqdm
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
for _ in tqdm(range(total_frames), desc="Processing Frames", unit="frames"):
    ret, frame = cap.read()
    if not ret:
        break

    with torch.no_grad():
        outputs = predictor(frame)['instances']

    results = DensePoseResultExtractor()(outputs)
    cmap = cv2.COLORMAP_VIRIDIS
    arr = cv2.applyColorMap(np.zeros((height, width), dtype=np.uint8), cmap)
    out_frame = Visualizer(alpha=1, cmap=cmap).visualize(arr, results)        
    out.write(out_frame)

``

[tadeodonegana]

Hi @JD-2006,

Thanks for the suggestion. Please feel free to open a Pull Request with this feature and we will be happy to evaluate and merge it!

[JD-2006]

Cheers! Will do.