fix: change model path in inference test

9b13ea03 · Cresson Remi · 10893d92 · 9b13ea03 · 9b13ea03
Commit 9b13ea03 authored 6 months ago by Cresson Remi
Hide whitespace changes
Inline Side-by-side

Showing

with 4 additions and 4 deletions
+4 -4
--- a/Dockerfile
+++ b/Dockerfile
@@ -19,8 +19,8 @@ COPY docker/requirements.txt /tmp/requirements.txt
 RUN python3 -m pip install -r /tmp/requirements.txt

 # Build remote modules
-RUN cd /src/otb/otb/Modules/Remote/ && git clone https://gitlab.irstea.fr/remi.cresson/SimpleExtractionTools.git
-RUN cd /src/otb/otb/Modules/Remote/ && git clone https://gitlab.irstea.fr/remi.cresson/mlutils.git
+RUN cd /src/otb/otb/Modules/Remote/ && git clone https://forgemia.inra.fr/orfeo-toolbox/otb-simpleextractiontools.git
+RUN cd /src/otb/otb/Modules/Remote/ && git clone https://forgemia.inra.fr/orfeo-toolbox/otb-mlutils.git
 COPY . /src/otb/otb/Modules/Remote/decloud/
 RUN cd /src/otb/build/OTB/build && cmake /src/otb/otb/ -DModule_SimpleExtractionTools=ON -DModule_MLUtils=ON -DBUILD_TESTING=OFF -DModule_OTBDecloud=ON 
 RUN cd /src/otb/build/OTB/build && make -j $(nproc --all) install

--- a/tests/inference_unittest.py
+++ b/tests/inference_unittest.py
@@ -27,7 +27,7 @@ class InferenceTest(DecloudTest):
        baseline_path = self.get_path("baseline/reconstructed_baseline_w_mosaic.tif")

        # Model
-        model_path = self.get_path("models/crga_os2david_occitanie_pretrained")
+        model_path = self.get_path("models/crga_os2_david_116-migration-tensorflow-2_466a1_bt32_bv32_lr7e-05_40gpus")

        # Input sources
        s1_tm1 = [
@@ -93,7 +93,7 @@ class InferenceTest(DecloudTest):
        baseline_path = self.get_path("baseline/reconstructed_baseline_w_preprocessor.tif")

        # Model
-        model_path = self.get_path("models/crga_os2david_occitanie_pretrained")
+        model_path = self.get_path("models/crga_os2_david_116-migration-tensorflow-2_466a1_bt32_bv32_lr7e-05_40gpus")

        # Input sources
        s1_tm1 = [