Skip to content
GitLab

Checkpoints callbacks fixes

Merged Cresson Remi requested to merge checkpoints_callbacks_fixes into master
Viewing commit d604263c
Prev
Show latest version
2 files
+ 16
− 11
tests/decloud_unittest.py
+ 4
− 1
@@ -15,7 +15,10 @@ class DecloudTest(ABC, unittest.TestCase):
DECLOUD_DATA_DIR = get_env_var("DECLOUD_DATA_DIR")
def get_path(self, path):
return os.path.join(self.DECLOUD_DATA_DIR, path)
pth = os.path.join(self.DECLOUD_DATA_DIR, path)
if not os.path.exists(pth):
raise FileNotFoundError(f"Directory {pth} not found!")
return pth
def compare_images(self, image, reference, mae_threshold=0.01):
tests/train_from_tfrecords_unittest.py
+ 12
− 10
@@ -8,6 +8,7 @@ from .decloud_unittest import DecloudTest
SAVEDMODEL_FILENAME = "saved_model.pb"
def is_savedmodel_written(args_list):
out_savedmodel = "/tmp/savedmodel"
base_args = ["--logdir", "/tmp/logdir",
@@ -25,8 +26,9 @@ def is_savedmodel_written(args_list):
OS2_TFREC_PTH = "baseline/TFRecord/CRGA"
OS2_ALL_BANDS_TFREC_PTH = "/baseline/TFRecord/CRGA_all_bands"
MERANER_ALL_BANDS_TFREC_PTH = "/baseline/TFRecord/CRGA_all_bands"
OS2_ALL_BANDS_TFREC_PTH = "baseline/TFRecord/CRGA_all_bands"
MERANER_ALL_BANDS_TFREC_PTH = "baseline/TFRecord/CRGA_all_bands"
ERRMSG = f"File {SAVEDMODEL_FILENAME} not found !"
class TrainFromTFRecordsTest(DecloudTest):
@@ -34,42 +36,42 @@ class TrainFromTFRecordsTest(DecloudTest):
def test_trainFromTFRecords_os1_unet(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(OS2_TFREC_PTH),
"--model", "crga_os1_unet"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
def test_trainFromTFRecords_os2_david(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(OS2_TFREC_PTH),
"--model", "crga_os2_david"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
def test_trainFromTFRecords_os2_unet(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(OS2_TFREC_PTH),
"--model", "crga_os2_unet"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
def test_trainFromTFRecords_os1_unet_all_bands(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(OS2_ALL_BANDS_TFREC_PTH),
"--model", "crga_os1_unet_all_bands"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
def test_trainFromTFRecords_os2_david_all_bands(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(OS2_ALL_BANDS_TFREC_PTH),
"--model", "crga_os2_david_all_bands"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
def test_trainFromTFRecords_os2_unet_all_bands(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(OS2_ALL_BANDS_TFREC_PTH),
"--model", "crga_os2_unet_all_bands"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
def test_trainFromTFRecords_meraner_unet(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(MERANER_ALL_BANDS_TFREC_PTH),
"--model", "meraner_unet"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
def test_trainFromTFRecords_meraner_unet_all_bands(self):
self.assertTrue(is_savedmodel_written(["--training_record", self.get_path(MERANER_ALL_BANDS_TFREC_PTH),
"--model", "meraner_unet_all_bands"]),
"File {} not found !".format(SAVEDMODEL_FILENAME))
ERRMSG)
if __name__ == '__main__':