Tensorflow剛開始訓練killed bug解決方案

這個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)