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

FROM tensorflow/tensorflow:1.15.4

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

COPY ./lib/requirements.txt /home

RUN pip install -r /home/requirements.txt

ENV PYTHONPATH "/home/lib"

COPY ./lib /home/lib
WORKDIR /home/work
COPY examples/federated_learning/surface_defect_detection/aggregation_worker/  /home/work/

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