update remove bed script

mu_map/data/remove_bed.py

@@ -145,7 +145,7 @@ if __name__ == "__main__":
         while True:
             # compute image to display
             to_show = mu_map[_slice]
-            to_show = to_grayscale(to_show)
+            to_show = to_grayscale(to_show, min_val=mu_map.min(), max_val=mu_map.max())
             to_show = cv.resize(to_show, (window_size, window_size))
 
             if visual_mode == VisualMode.DRAW_CONTOURS:
@@ -178,10 +178,11 @@ if __name__ == "__main__":
             key = cv.waitKey(100)
             if key == ord("q"):
                 # write current contours to output file
-                scale_points(points, 1.0 / scale)
-                bed_contours[str(_id)] = points.copy()
-                with open(args.output_file, mode="w") as f:
-                    f.write(json.dumps(bed_contours, sort_keys=True))
+                if len(points) > 0:
+                    scale_points(points, 1.0 / scale)
+                    bed_contours[str(_id)] = points.copy()
+                    with open(args.output_file, mode="w") as f:
+                        f.write(json.dumps(bed_contours, sort_keys=True))
                 exit(0)
             elif key == ord("d"):
                 points = points[:-1]