Tensorflow刚开始训练killed bug解决方案

最新推荐文章于 2024-08-23 08:49:09 发布

RanCh.Wong

最新推荐文章于 2024-08-23 08:49:09 发布

阅读量5.3k

点赞数 2

分类专栏： python

本文链接：https://blog.csdn.net/qq_34192019/article/details/101020930

版权

python 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

这个BUG弄了整整两天，记录一下
在这里插入图片描述
各种百度，各种换模型，搞得头昏脑胀，最后通过对比实验，监测到killed之前内存暴涨，怀疑是内存泄漏或者内存不够，原来的训练图片尺寸5472×3648,resize一下就可以正常训练了，resize代码如下

import os
from PIL import Image
 
filename = os.listdir("/home/wxy/下载/VOCdevkit/VOC2007/JPEGImages/")
base_dir = "/home/wxy/下载/VOCdevkit/VOC2007/JPEGImages/"
new_dir  = "/home/wxy/下载/VOCdevkit/VOC2007/JPEGImages919/"
size_m = 450
size_n = 300
 
for img in filename:
    image = Image.open(base_dir + img)
    image_size = image.resize((size_m, size_n),Image.ANTIALIAS)
    image_size.save(new_dir+ img)