ARG METRIC_IMAGE=None
FROM ${METRIC_IMAGE} as base

#COPY --from=vg-code.gml-team.ru:4567/framework/attacks/datasets/coco:25e_shu /data /

RUN apt-get update -q 
RUN apt-get install build-essential -qqy
RUN pip3 install opencv-python
RUN pip3 install numpy
RUN pip3 install tqdm
RUN pip3 install av
RUN pip3 install scipy
RUN pip3 install scikit-image
RUN pip3 install PyWavelets
RUN pip3 install IQA_pytorch
RUN pip3 install frozendict
RUN pip3 install lpips
RUN pip3 install torchmetrics
RUN pip3 install pytorch-wavelets
RUN pip3 install pandas
RUN pip3 install scikit-learn
RUN pip3 install pyiqa
RUN pip3 install pytorch-msssim
RUN pip3 install kornia

COPY ./metrics.py /metrics.py
COPY ./main_eval_script.py /main_eval_script.py
COPY ./setup_modules.py /setup_modules.py
COPY ./config_parser.py /config_parser.py
COPY ./quality_evaluators.py /quality_evaluators.py
COPY ./helpers.py /helpers.py
COPY ./dataloaders.py /dataloaders.py
COPY ./raw_data_scheme.py /raw_data_scheme.py
COPY ./color_transforms_255.py /color_transforms_255.py
COPY ./attack_presets_codecs.json /attack_presets_codecs.json
COPY ./defended_model.py /defended_model.py
COPY ./defence_scoring_methods.py /defence_scoring_methods.py
COPY ./codec_losses.py ./codec_losses.py
COPY ./codec_scoring_methods.py ./codec_scoring_methods.py
COPY ./traditional_reference_codec.py /traditional_reference_codec.py

COPY run.py /run.py

