前提
Google colabで画像の合成を行い、Google Driveに保存するプログラムを作っています。
しかし、RAMをかなり消費してしまうようでエラーが出てしまいます。
BicubicをNEARESTに変えたりしましたが結果は同じでした。
このセルの前に
combined_I = []
FG_D = []
BG_D = []
GT_D = []
Trimap_D = [] を作成済みになります。
発生している問題・エラーメッセージ
RAMを全て消費し、セッションがクラッシュしました。
該当のソースコード
python
for i,(FG,GT,Trimap) in enumerate(zip(FG0,GT0,Trimap0)): fg = np.array(FG, np.float32) trimap = np.array(Trimap, np.float32) h,w = fg.shape[:2] for j,BG in enumerate(BG0): alpha = np.zeros((h, w, 1), np.float32) alpha[:, :, 0] = GT / 255 bg = np.array(BG,np.float32) bg = Image.fromarray(np.uint8(bg)) bg = np.asarray(bg.resize((w,h),Image.Bicubic)) comp = alpha * fg + (1 - alpha) * bg comp = np.asarray(comp)#PIL 画像を numPy 配列に変換 comp = comp.astype(np.uint8) combined_I.append(comp) FG_D.append(FG) BG_D.append(BG) GT_D.append(GT) Trimap_D.append(Trimap)
試したこと
BicubicをNEARESTに変更
補足情報
Google Colabは無料版です。
0 コメント