imageName = re.findall('[^/]+$', self.fileName[0]) name, _ = str(imageName[0]).split('.') FOM_Module = hub.Module(name="first_order_motion") FOM_Module.generate(source_image="{}".format(self.fileName[0]), # 输入图像 driving_video="{}".format(self.fileName[1]), # 输入驱动视频 ratio=0.4, image_size=256, output_dir=f'{self.output_dir}/', # 输出文件夹 filename=f'{name}.mp4', # 输出文件名 use_gpu=False)实时显示此段代码进度
时间: 2023-12-19 18:07:03 浏览: 56
如果你想实时显示代码进度,可以使用tqdm库。你可以在代码中添加以下代码:
```
from tqdm import tqdm
# 将FOM_Module.generate()函数包装在tqdm()中
for _ in tqdm(FOM_Module.generate(source_image="{}".format(self.fileName[0]), # 输入图像
driving_video="{}".format(self.fileName[1]), # 输入驱动视频
ratio=0.4,
image_size=256,
output_dir=f'{self.output_dir}/', # 输出文件夹
filename=f'{name}.mp4', # 输出文件名
use_gpu=False)):
pass
```
这将在命令行中以进度条的形式显示代码运行进度。
阅读全文