From 9b13ea032927fa25261e7eb342081c8997507de3 Mon Sep 17 00:00:00 2001
From: Cresson Remi <remi.cresson@irstea.fr>
Date: Tue, 3 Sep 2024 11:38:58 +0200
Subject: [PATCH] fix: change model path in inference test
---
Dockerfile | 4 ++--
tests/inference_unittest.py | 4 ++--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/Dockerfile b/Dockerfile
index 31f4979..8bc8cea 100644
--- 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
diff --git a/tests/inference_unittest.py b/tests/inference_unittest.py
index 316b9c4..b45348b 100644
--- 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 = [
--
GitLab