Merge pull request #36 from naufalso/patch-1

Update make_dataset.py for skiping non .binary
google · Apr 30, 2023 · a668dca · a668dca
2 parents 3d2d214 + bea37df
commit a668dca
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/data_gen/merl/make_dataset.py b/data_gen/merl/make_dataset.py
@@ -72,6 +72,9 @@ def main(_):
 
     brdf_paths = xm.os.sortglob(FLAGS.indir)
     for i, path in enumerate(tqdm(brdf_paths, desc="Training & Validation")):
+        if not path.endswith('.binary'):
+            continue
+
         brdf = MERL(path=path)
 
         rusink = brdf.tbl[:, :3]