numpyでオーグメンテーションその1 flip

画像のオーグメンテーションでのnumpyを使えばflipが一行で出来るのを知らない人がいたので紹介。

a = np.zeros(5*5).reshape(1,5,5)
a[0,1,1] = 1
# vertical flip
a = np.append(a,a[:,::-1,:],axis=0)
# horizon flip
a = np.append(a,a[:,:,::-1],axis=0)

# 確認
print(a)

これだけで垂直&水平&垂直水平の4倍に倍増

やったね! 動かして確認してみましょう