【无标题】安装MMCV报错and module ‘mmcv‘ has no attribute ‘ProgressBar‘

让你安装visual studio…
不要安装!!!!
参考这个博主
https://blog.csdn.net/m0_73914600/article/details/127176961
安装成功后
使用ProgressBar时报错
module ‘mmcv’ has no attribute ‘ProgressBar’
请用 tqdm库代替

imgs = mmcv.VideoReader(input_video)

prog_bar = mmcv.ProgressBar(len(imgs))

# 对视频逐帧处理
for frame_id, img in enumerate(imgs):
    
    ## 处理单帧画面
    img = pred_single_frame(img, show_class_id=None)
    # 将处理后的该帧画面图像文件,保存至 /tmp 目录下
    img.save(f'{temp_out_dir}/{frame_id:06d}.jpg', "BMP")
    
    prog_bar.update() # 更新进度条

# 把每一帧串成视频文件
mmcv.frames2video(temp_out_dir, output_path, fps=imgs.fps, fourcc='mp4v')

shutil.rmtree(temp_out_dir) # 删除存放每帧画面的临时文件夹
print('删除临时文件夹', temp_out_dir)
print('视频已生成', output_path)

改为

import mmcv
from tqdm import tqdm
import shutil
from PIL import Image  # 假设 `pred_single_frame` 返回的是 PIL Image 对象

input_video = 'input.mp4'
temp_out_dir = '/tmp'
output_path = 'output.mp4'

# 读取视频
imgs = mmcv.VideoReader(input_video)

# 使用 tqdm 显示进度条
with tqdm(total=len(imgs), desc="Processing frames") as pbar:
    # 对视频逐帧处理
    for frame_id, img in enumerate(imgs):
        ## 处理单帧画面
        img = pred_single_frame(img, show_class_id=None)
        
        # 将处理后的该帧画面图像文件,保存至 /tmp 目录下
        img.save(f'{temp_out_dir}/{frame_id:06d}.jpg', "BMP")
        
        pbar.update(1)  # 更新进度条

# 把每一帧串成视频文件
mmcv.frames2video(temp_out_dir, output_path, fps=imgs.fps, fourcc='mp4v')

# 删除存放每帧画面的临时文件夹
shutil.rmtree(temp_out_dir)
print('删除临时文件夹', temp_out_dir)
print('视频已生成', output_path)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值