sedna/examples/federated-learning-surface-defect-detection-train.Dockerfile

FROM tensorflow/tensorflow:2.3.0

RUN apt update \
  && apt install -y libgl1-mesa-glx

COPY ./lib/requirements.txt /home
RUN pip install -r /home/requirements.txt
RUN pip install keras~=2.4.3
RUN pip install opencv-python==4.4.0.44
RUN pip install Pillow==8.0.1

ENV PYTHONPATH "/home/lib"

COPY ./lib /home/lib

WORKDIR /home/work
COPY examples/federated_learning/surface_defect_detection/training_worker/ /home/work/

ENTRYPOINT ["python", "train.py"]