Update train.py

lyf6 · Mar 21, 2021 · d8cf149 · d8cf149
1 parent 39d2b2f
commit d8cf149
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -115,11 +115,11 @@ def validation(self):
                     for i in range(out1.shape[0]):
                         mask = Image.fromarray(out1[i].astype(np.uint8), mode="P")
                         mask.putpalette(cmap)
-                        mask.save("outdir/masks/train/im1/" + id[i])
+                        mask.save("outdir/masks/val/im1/" + id[i])
 
                         mask = Image.fromarray(out2[i].astype(np.uint8), mode="P")
                         mask.putpalette(cmap)
-                        mask.save("outdir/masks/train/im2/" + id[i])
+                        mask.save("outdir/masks/val/im2/" + id[i])
 
                 metric.add_batch(out1, mask1.numpy())
                 metric.add_batch(out2, mask2.numpy())