add latest changes

Dockerfile

@@ -12,13 +12,8 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
   
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
   
-# RUN echo $(ls /run/secrets/)
-
-
 WORKDIR /code
 
-# COPY ./requirements.txt /code/requirements.txt
-
 # add conda
 RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -P /code/
 RUN chmod 777 /code/Miniconda3-latest-Linux-x86_64.sh
@@ -53,45 +48,31 @@ RUN pip3 install setuptools-rust
 
 RUN conda install pillow -y
 
-# RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
-
-RUN conda install -c pytorch pytorch==1.7.1 torchvision==0.8.2 
-# RUN pip install torchvision-cpu==0.8.2
-
-# WORKDIR /home/user
-
-# RUN git clone https://github.com/open-mmlab/mim.git
+RUN pip3 install torch==1.11.0+cu115 torchvision==0.12.0+cu115 --extra-index-url https://download.pytorch.org/whl/cu115
 
 RUN pip3 install openmim
 
-
 RUN conda install -c conda-forge gradio -y 
 
- 
-# RUN --mount=type=secret,id=git_token,mode=0444,required=true \
-#     echo $(https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git) 
-#     # git clone https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git
-
 WORKDIR /home/user
 
 RUN --mount=type=secret,id=git_token,mode=0444,required=true \
-    git clone https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git
+    git clone --branch mmseg-only https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git
 
 
 WORKDIR hls-foundation-os 
 
-RUN pip3 install fine-tuning-examples/
-
+RUN pip3 install -e .
 
-# RUN --mount=type=secret,id=git_token,mode=0444,required=true \
-#     pip3 install git+https://$(cat /run/secrets/git_token)@github.com/NASA-IMPACT/hls-foundation-os.git@mmseg-only
-
-RUN mim install mmcv-full==1.5.0
+RUN mim install mmcv-full==1.6.2 -f https://download.openmmlab.com/mmcv/dist/11.5/1.11.0/index.html
 
+RUN pip3 install rasterio scikit-image
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
+ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/code/miniconda/lib"
+
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["python3", "app.py"]
+CMD ["python3", "app.py"] 

app.py

@@ -184,7 +184,9 @@ def process_test_pipeline(custom_test_pipeline, bands=None):
     
     return custom_test_pipeline
 
-model = init_segmentor(config_path, ckpt, device='cpu')
+config = Config.fromfile(config_path)
+config.model.backbone.pretrained=None
+model = init_segmentor(config, ckpt, device='cpu')
 custom_test_pipeline=process_test_pipeline(model.cfg.data.test.pipeline, None)
 
 func = partial(inference_on_file, model=model, custom_test_pipeline=custom_test_pipeline)