使用glob.glob标记图像

Question

我有/ dataset /目录和14个子文件夹，即/ dataset / 0 /，dataset / 1 /等。在这些文件夹中，我有.jpg图像。 我必须标记这些图像，所以我写道：

cat_dog_train_path = './dataset/*/*.jpg'
# read addresses and labels from the 'train' folder
addrs = glob.glob(cat_dog_train_path)
labels = [0 if '0' in addr
          else 1 if '1' in addr
          else 2 if '2' in addr
          else 3 if '3' in addr
          else 4 if '4' in addr
          else 5 if '5' in addr
          else 6 if '6' in addr
          else 7 if '7' in addr
          else 8 if '8' in addr
          else 9 if '9' in addr
          else 10 if 'div' in addr
          else 11 if 'times' in addr
          else 12 if 'plus' in addr
          else 13 for addr in addrs]

如果我做对了，有人可以检查一下吗？当然可以帮助纠正我的代码。