Update train_util.py

library/train_util.py

@@ -905,6 +905,10 @@ class FineTuningDataset(BaseDataset):
         # path情報を作る
         if os.path.exists(image_key):
           abs_path = image_key
+        else:
+          npz_path = os.path.join(glob.escape(train_data_dir), image_key + ".npz")
+          if os.path.exists(npz_path):
+            abs_path = npz_path
           else:
             # わりといい加減だがいい方法が思いつかん
             abs_path = glob_images(subset.image_dir, image_key)