Pythonを使って画像を分割・保存したいが、分割した画像が真っ黒になってしまう（IndexError: cannot do a non-empty take from an empty axes.）

Question

前提・実現したいこと

Pythonの初心者です。
フルカラーの画像を細かく分割して保存したいのですが、分割して保存した画像の色が失われてしまいます。
画像の分割、保存はできているようですが、どこかの段階で画素の値が全て0 or 小さい値になってしまっている疑いがあります。
画像の色を保たれたまま保存するにはどうすればよいか、ご助言をいただけませんでしょうか？

発生している問題・エラーメッセージ

IndexError: cannot do a non-empty take from an empty axes.

該当のソースコード

Pycharmで以下のように記述しました。

from skimage import io
import os
import skimage

tile_size = 100
filename = 'pic.png'

new_dir = 'data/'
# os.mkdir(new_dir)

im = io.imread(filename)

print(im)
for i in range(10):
   ii = tile_size * i + 100
   for j in range(10):
       jj = tile_size * j + 100
       tmp = im[ii:ii+tile_size, jj:jj+tile_size, 1]
       print(tmp)
       # for z in range(tmp.shape[0]):
       #     tmp[z, :, :] = skimage.filters.median(tmp[z, :, :])
       io.imsave(new_dir + str(i)+ "_" + str(j) + ".tif", tmp)

試したこと

上記と同じエラーメッセージが出た方の対処法を調べたのですが、エラーの原因、対処法を特定できませんでした。

補足情報（FW/ツールのバージョンなど）

Answer 1

以下のようにするとできると思います。

OpenCV - 画像をグリッド上に分割する方法について

サンプルコード

画像を 100x100 のグリッド上に分割して output ディレクトリ以下に保存する例

from pathlib import Path

import numpy as np
from skimage import io

img = io.imread("sample.jpg")

vsize, hsize = 100, 100  # 分割後の各セルのサイズ
h, w = img.shape[:2]  # 画像の大きさ
num_vsplits, num_hsplits = np.floor_divide([h, w], [vsize, hsize])  # 分割数

# 均等に分割できないと np.spllt() が使えないので、
# 除算したときに余りがでないように画像の端数を切り捨てる。
crop_img = img[: num_vsplits * vsize, : num_hsplits * hsize]

# 分割する。
grid = []
for h_img in np.vsplit(crop_img, num_vsplits):  # 垂直方向に分割する。
    for v_img in np.hsplit(h_img, num_hsplits):  # 水平方向に分割する。
        grid.append(v_img)

out_dir = Path("output")  # 出力ディレクトリ
out_dir.mkdir(exist_ok=True)

for i, small_img in enumerate(grid):
    out_path = out_dir / f"{i}.png"
    io.imsave(out_path, small_img)

Comments

[masa_jl033]

ありがとうございました。できました。私が投稿した記述では何がいけなかったのでしょうか？

[tiitoi]

`im[ii : ii + tile_size, jj : jj + tile_size, 1]` としているので、RGB のうち Green のチャンネルだけスライスされているため、出力は1チャンネルのグレースケールとなります。
RGB のままスライスしたい場合は im[ii : ii + tile_size, jj : jj + tile_size] でいいのではないでしょうか。

[masa_jl033]

ご教示くださりありがとうございます。元のコードが３次元画像を扱っていたので、その分はz軸方向のパラメータを入れていたようです。次元ではなく、RGBを表していたのですね。