linux环境下经常内存溢出的一种解决办法

首先说一下,内存经常溢出,爆内存不一定是因为batch_size过大,也有可能是模型没有加载上。弄了一天才把这个问题解决

问题:
我在使用paddlehub的下面这段函数时,经常爆内存

paddlehub.Module(name='deeplabv3p_xception65_humanseg')

简单说一下过程

 //我先用humanseg加载模型,这个deeplabv3p_xception65_humanseg
 //模型如果Linux环境未安装,则自动安装
 humanseg = paddlehub.Module(name='deeplabv3p_xception65_humanseg')
 //开始遍历frames下面的所有图片
 files = [frames + i for i in os.listdir(frames)]
 //开始抠图
 humanseg.segmentation(data={'image': files})  # 抠图

其实在这里,我就开始内存飙升,而且程序直接被killed掉

 humanseg = paddlehub.Module(name='deeplabv3p_xception65_humanseg')

我就当时在想是不是batch_size是不是过大。这时候我就又开始只扣一张图片,不是扣一个视频。可是依然不行,我这时候就思考是不是.paddlehub下面的deeplabv3p_xception65_humanseg模型没有加载成功。
这时候我就把我电脑里这个模型打包成zip上传到linux环境下,然后解压到指定

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值