如何在linux环境下优化python程序对内存的使用

1.经常在linux环境下，程序因为使用较大的内存被kill掉，所以我想问一下是否有方法限制python对内存的使用，比如只让他获得2个g的内存使用。

2.错误代码：

        //加载抠图模型（用的飞桨的库）
        humanseg = hub.Module(name='deeplabv3p_xception65_humanseg')
        //下面两句开始遍历frames文件夹下的图片，然后开始抠图
        files = [frames + i for i in os.listdir(frames)]
    humanseg.segmentation(data={'image': files})  # 抠图

https://www.jb51.net/article/65833.htm