class_names = ['mountain', 'street', 'glacier', 'buildings', 'sea', 'forest'] class_names_label = {i:class_name for i,class_name in enumerate(class_names)} def load_data(): datasets = ['../input/seg_train/seg_train', '../input/seg_test/seg_test'] output = [] for dataset in datasets: images = [] labels = [] for folder in os.listdir(dataset): label = class_names_label[folder]
class_names_label[folder] という点の意味が全く理解できません。
os.listdir('../input/seg_train/seg_train/')の結果は
['mountain', 'street', 'buildings', 'sea', 'forest', 'glacier']
です。
上の結果から考えると、label=class_names_label['mountain']ということになるのでしょうか?そんなコードはあり得ないですよね。
どなたかお詳しい方、ご教授下さいませ。
あなたの回答
tips
プレビュー