Spaces:

abhishekrs4
/

Oil_Spill_Segmentation

Runtime error

abhishekrs4 commited on Feb 13

Commit

5be67d1

•

1 Parent(s): 5f62bb5

fixed import in compute stats for train script

Files changed (1) hide show

training/compute_stats_for_train.py CHANGED Viewed

@@ -4,7 +4,7 @@ import argparse
 import numpy as np
 from skimage.io import imread
-from logger_utils import write_json_file
 def compute_stats(FLAGS):
     list_images = sorted(os.listdir(FLAGS.dir_images))

 import numpy as np
 from skimage.io import imread
+from logger_utils import write_dict_to_json
 def compute_stats(FLAGS):
     list_images = sorted(os.listdir(FLAGS.dir_images))