【セグメンテーション】ラベルデータをインデックスカラーのpngファイルとして保存・読込みする - Qiita

はじめに ニューラルネットを用いた機械学習の中でも、セグメンテーションは1ピクセルごとのラベルが必要になることから、ラベルのインプットを工夫しないと学習のボトルネックになることがあります。 整形してあるデータセットであれば良いですが、KaggleなどはCSVファイルでラベルが配布される場合がある為、学習時に都度CSVからラベルを作成してモデルにインプットした場合、学習が非常に遅くなります。 そこで予め全てのラベルをインデックスカラーのpngファイルとして保存しておき、学習時には読込みだけを行う方法を調べたので、備忘録として残します。 環境 Google Colaboratory Pro コード モジュールのimport import numpy as np from PIL import Image import torch from matplotlib import pyplot as