使用gpu抠图生成的是空白图片没有人像,但用cpu抠生存的图片里就有人像
收藏
用的同样的代码
0
收藏
请登录后评论
use_gpu为true,生成的图片是空白没有人像,use_gpu改成false生成的图片就有人像,试了不同的图片都是这样
human_seg = hub.Module(name="deeplabv3p_xception65_humanseg")
result = human_seg.segmentation(images=[cv2.imread('./frames/0.jpg')], visualization=True, use_gpu=True, output_dir='./humanseg_output/')