- カラー画像の場合、入力レイヤー数は、RGBの3種類になります。今はグレイスケー ル画像なので入力レイヤー数は1です。また、1枚の画像から2枚の画像を出力するの で、出力レイヤー数は2です。 def edge_filter(): filter0 = np.array( [[ 2, 1, 0,-1,-2], [ 3, 2, 0,-2,-3], [ 4, 3, 0,-3,-4], [ 3, 2, 0,-2,-3], [ 2, 1, 0,-1,-2]]) / 23.0 filter1 = np.array( [[ 2, 3, 4, 3, 2], [ 1, 2, 3, 2, 1], [ 0, 0, 0, 0, 0], [-1,-2,-3,-2,-1], [-2,-3,-4,-3,-2]]) / 23.0 filter_array = np.zeros([5,5,1,2]) filter_array[:,:,0,0] = filter0 filter_array[:,:,0,1] = filter1 return tf.constant(filter_array, dtype=tf.float32) それぞれのフィルターを 2次元リストとして用意 すべてのフィルターを格納 する多次元リストを用意 該当部分にフィルターの値を格納 定数オブジェクトに 変換して返却 縦/横のエッジを抽出する フィルターを定義する関数