Point at 46.6% model

SkyTruth · Sep 5, 2024 · cb6ed8e · cb6ed8e
1 parent 6072c51
commit cb6ed8e
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/alembic/versions/c941681a050d_add_initial_records.py b/alembic/versions/c941681a050d_add_initial_records.py
@@ -101,15 +101,15 @@ def upgrade() -> None:
             ),
             database_schema.Model(
                 type="FASTAIUNET",
-                file_path="experiments/2024_08_17_11_03_50_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt",
+                file_path="experiments/2024_09_04_21_34_24_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt",
                 layers=["VV"],
                 cls_map={
                     0: "BACKGROUND",
                     1: "INFRA",
                     2: "NATURAL",
                     3: "VESSEL",
                 },  # inference_idx maps to class table
-                name="ResNet34 44.5%",
+                name="ResNet34 46.6%",
                 tile_width_m=40844,  # Used to calculate zoom
                 tile_width_px=512,  # Used to calculate scale
                 epochs=500,
@@ -122,7 +122,7 @@ def upgrade() -> None:
                     "groundtruth_dice_thresh": 0.0,
                 },
                 backbone_size=34,
-                pixel_f1=0.495,
+                pixel_f1=0.532,
                 # instance_f1=0.0, # TODO CALCULATE
             ),
         ]

diff --git a/images/stack_config/Pulumi.prod20240903.yaml b/images/stack_config/Pulumi.prod20240903.yaml
@@ -2,7 +2,7 @@ encryptionsalt: v1:R2sLsAf/bQc=:v1:hBVgdX8qN8+5MyEa:oJlozt2wlaT51vy2RpgoKZG/PVUu
 config:
   aws:region: eu-central-1
   cerulean-cloud-images:weights_bucket: ceruleanml
-  cerulean-cloud-images:weights_name: experiments/2024_08_17_11_03_50_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt
+  cerulean-cloud-images:weights_name: experiments/2024_09_04_21_34_24_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt
   gcp:project: cerulean-338116
   gcp:region: europe-west1
   gcp:zone: europe-west1-b
diff --git a/images/stack_config/Pulumi.staging.yaml b/images/stack_config/Pulumi.staging.yaml
@@ -2,7 +2,7 @@ encryptionsalt: v1:5bZrZv+vt08=:v1:2g/MGwZygH99RO1o:/SYBjfvFWKhefeTSbupB2bXA3G8w
 config:
   aws:region: eu-central-1
   cerulean-cloud-images:weights_bucket: ceruleanml
-  cerulean-cloud-images:weights_name: experiments/2024_08_17_11_03_50_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt
+  cerulean-cloud-images:weights_name: experiments/2024_09_04_21_34_24_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt
   gcp:project: cerulean-338116
   gcp:region: europe-west1
   gcp:zone: europe-west1-b
diff --git a/images/stack_config/Pulumi.test.yaml b/images/stack_config/Pulumi.test.yaml
@@ -2,7 +2,7 @@ encryptionsalt: v1:/dV9M1NH8EA=:v1:IPmnUfz0q6E6S8y7:fl72KMTy2rm7uY4bZzP4p790z9k8
 config:
   aws:region: eu-central-1
   cerulean-cloud-images:weights_bucket: ceruleanml
-  cerulean-cloud-images:weights_name: experiments/2024_08_17_11_03_50_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt
+  cerulean-cloud-images:weights_name: experiments/2024_09_04_21_34_24_4cls_resnet34_pr512_px1024_100epochs_unet/tracing_cpu_model.pt
   gcp:project: cerulean-338116
   gcp:region: europe-west1
   gcp:zone: europe-west1-b