@@ -74,20 +74,13 @@ RUN apt update -q=2 \
74
74
&& apt-get update -q=2 \
75
75
&& apt-get install -y --no-install-recommends cmake=3.28.3* cmake-data=3.28.3*
76
76
77
- # Add inception_graphdef model to example repo
77
+ # Add densenet_onnx model to example repo
78
78
# FIXME: This should be changed to using the fetch_models.sh script
79
79
# in order to ensure the public facing docs are up-to-date.
80
80
WORKDIR /workspace/docs/examples/model_repository
81
- RUN mkdir -p model_repository/inception_onnx/1 && \
82
- wget -O /tmp/inception_v3_2016_08_28_frozen.pb.tar.gz \
83
- https://storage.googleapis.com/download.tensorflow.org/models/inception_v3_2016_08_28_frozen.pb.tar.gz && \
84
- (cd /tmp && tar xzf inception_v3_2016_08_28_frozen.pb.tar.gz) && \
85
- python3 -m venv tf2onnx && \
86
- source ./tf2onnx/bin/activate && \
87
- pip3 install "numpy<2" tensorflow tf2onnx && \
88
- python3 -m tf2onnx.convert --graphdef /tmp/inception_v3_2016_08_28_frozen.pb --output inception_v3_onnx.model.onnx --inputs input:0 --outputs InceptionV3/Predictions/Softmax:0 && \
89
- deactivate && \
90
- mv inception_v3_onnx.model.onnx model_repository/inception_onnx/1/model.onnx
81
+ RUN mkdir -p densenet_onnx/1 && \
82
+ wget -O densenet_onnx/1/model.onnx \
83
+ https://github.com/onnx/models/raw/main/validated/vision/classification/densenet-121/model/densenet-7.onnx
91
84
92
85
# Update the qa/ directory with test executables, models, etc.
93
86
WORKDIR /workspace
@@ -117,7 +110,7 @@ RUN mkdir -p qa/common && \
117
110
cp -r docs/examples/model_repository/simple_identity qa/L0_grpc/models && \
118
111
cp -r docs/examples/model_repository/simple_sequence qa/L0_grpc/models && \
119
112
cp -r docs/examples/model_repository/simple_string qa/L0_grpc/models && \
120
- cp -r docs/examples/model_repository/inception_onnx qa/L0_grpc/models && \
113
+ cp -r docs/examples/model_repository/densenet_onnx qa/L0_grpc/models && \
121
114
mkdir qa/L0_grpc_state_cleanup/models && \
122
115
cp -r /workspace/src/test/models/repeat_int32 qa/L0_grpc_state_cleanup/models/ && \
123
116
mkdir qa/L0_http/models && \
@@ -126,7 +119,7 @@ RUN mkdir -p qa/common && \
126
119
cp -r docs/examples/model_repository/simple_identity qa/L0_http/models && \
127
120
cp -r docs/examples/model_repository/simple_sequence qa/L0_http/models && \
128
121
cp -r docs/examples/model_repository/simple_string qa/L0_http/models && \
129
- cp -r docs/examples/model_repository/inception_onnx qa/L0_grpc /models && \
122
+ cp -r docs/examples/model_repository/densenet_onnx qa/L0_http /models && \
130
123
mkdir qa/L0_https/models && \
131
124
cp -r docs/examples/model_repository/simple qa/L0_https/models/. && \
132
125
mkdir qa/L0_secure_grpc/models && \
0 commit comments