DockerfileSeg

ARG PYTORCH="2.2.1"
ARG CUDA="12.1"
ARG CUDNN="8"

FROM docker.io/pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel

RUN apt-get update \
    && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

RUN python -m pip install --upgrade pip

WORKDIR /app
COPY lowAltitude_classification/requirements.txt /app/requirements.txt
RUN python -m pip install -r /app/requirements.txt
COPY lowAltitude_segmentation/Mask2Former/requirements.txt /app/requirements.txt
RUN python -m pip install -r /app/requirements.txt

RUN pip install torch torchvision
RUN pip install opencv-python
RUN pip install git+https://github.com/cocodataset/panopticapi.git
RUN pip install git+https://github.com/mcordts/cityscapesScripts.git
# Maybe in container?
RUN git clone https://github.com/facebookresearch/detectron2 /detectron2
RUN cd /detectron2 && pip install -e .

ENV PYTHONPATH=/app:$PYTHONPATH