Skip to content

Commit

Permalink
refactor tf serving for Raspberry
Browse files Browse the repository at this point in the history
  • Loading branch information
Baptiste O'Jeanson committed Mar 31, 2023
1 parent a6e5f06 commit 2ab1255
Showing 1 changed file with 8 additions and 6 deletions.
14 changes: 8 additions & 6 deletions edge_model_serving/tf_serving/Dockerfile.raspberrypi
View file
Original file line number Diff line number Diff line change
@@ -1,10 +1,12 @@
FROM --platform=linux/arm/v7 emacski/tensorflow-serving

WORKDIR /models
COPY ./inception /models/inception
COPY ./mobilenet_v1_640x640 /models/mobilenet_v1_640x640
COPY ./mask_classification_model /models/mask_classification_model
COPY ./marker_quality_control /models/marker_quality_control
COPY ./models.config /models/models.config

ENTRYPOINT ["tensorflow_model_server", "--port=8500", "--rest_api_port=8501", "--model_config_file=/models/models.config"]
COPY tf_serving/models.config /models/models.config
COPY models/tf/inception /models/tf/inception
COPY models/tf/mobilenet_v1_640x640 /models/tf/mobilenet_v1_640x640
COPY models/tf/mask_classification_model /models/tf/mask_classification_model
COPY models/tf/mask_classification_model /models/tf/mask_classification_model
COPY models/tf/marker_quality_control /models/tf/marker_quality_control

ENTRYPOINT ["tensorflow_model_server", "--port=8500", "--rest_api_port=8501", "--model_config_file=/models/models.config"]

0 comments on commit 2ab1255

Please sign in to comment.