Update batch_hed.py

clone of phillipi/pix2pix#191
JinProton · Apr 1, 2020 · c9f5826 · c9f5826
1 parent 3b8170b
commit c9f5826
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/edges/batch_hed.py b/scripts/edges/batch_hed.py
@@ -75,7 +75,7 @@ def parse_args():
     net.forward()
     fuse = net.blobs['sigmoid-fuse'].data[0][0, :, :]
     # get rid of the border
-    fuse = fuse[border:-border, border:-border]
+    fuse = fuse[(border+35):(-border+35), (border+35):(-border+35)]
     # save hed file to the disk
     name, ext = os.path.splitext(imgList[i])
     sio.savemat(os.path.join(args.hed_mat_dir, name + '.mat'), {'edge_predict': fuse})