Skip to content
This repository has been archived by the owner on Apr 18, 2023. It is now read-only.

Commit

Permalink
Merge pull request #353 from huningxin/resnet50
Browse files Browse the repository at this point in the history
[Example] use resnet50 v1 and v2
  • Loading branch information
huningxin authored Dec 1, 2018
2 parents 4778f20 + 7832c13 commit 803547b
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 10 deletions.
16 changes: 8 additions & 8 deletions examples/image_classification/main.js
Original file line number Diff line number Diff line change
Expand Up @@ -99,9 +99,9 @@ const mobilenet_v2_onnx = {
softmax: true,
}
};
const resnet18_v1_onnx = {
modelName: 'Resnet v1(Onnx)',
modelFile: './model/resnet18v1.onnx',
const resnet_v1_onnx = {
modelName: 'ResNet50 v1(Onnx)',
modelFile: './model/resnet50v1.onnx',
labelsFile: './model/labels1000.txt',
inputSize: [224, 224, 3],
outputSize: 1000,
Expand All @@ -115,9 +115,9 @@ const resnet18_v1_onnx = {
softmax: true,
}
};
const resnet18_v2_onnx = {
modelName: 'Resnet v2(Onnx)',
modelFile: './model/resnet18v2.onnx',
const resnet_v2_onnx = {
modelName: 'ResNet50 v2(Onnx)',
modelFile: './model/resnet50v2.onnx',
labelsFile: './model/labels1000.txt',
inputSize: [224, 224, 3],
outputSize: 1000,
Expand Down Expand Up @@ -155,8 +155,8 @@ function main(camera) {
inception_resnet_v2_tflite,
squeezenet_onnx,
mobilenet_v2_onnx,
resnet18_v1_onnx,
resnet18_v2_onnx,
resnet_v1_onnx,
resnet_v2_onnx,
inceptionv2_onnx,
];

Expand Down
4 changes: 2 additions & 2 deletions examples/image_classification/model/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@ Download the following model packages, then uncompress them if necessary and mov
6. [Inception Resnet V2(TFlite)](https://storage.googleapis.com/download.tensorflow.org/models/tflite/model_zoo/upload_20180427/inception_resnet_v2_2018_04_27.tgz)
7. [Squeezenet(Onnx)](https://s3.amazonaws.com/onnx-model-zoo/squeezenet/squeezenet1.1/squeezenet1.1.onnx)
8. [Mobilenet v2(Onnx)](https://s3.amazonaws.com/onnx-model-zoo/mobilenet/mobilenetv2-1.0/mobilenetv2-1.0.onnx)
9. [Resnet-18 v1(Onnx)](https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet18v1/resnet18v1.onnx)
10. [Resnet-18 v2(Onnx)](https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet18v2/resnet18v2.onnx)
9. [Resnet-50 v1(Onnx)](https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet50v1/resnet50v1.onnx)
10. [Resnet-50 v2(Onnx)](https://s3.amazonaws.com/onnx-model-zoo/resnet/resnet50v2/resnet50v2.onnx)
11. [Inception v2(Onnx)](https://s3.amazonaws.com/download.onnx/models/opset_9/inception_v2.tar.gz) (Untar it and rename `model.onnx` to `inceptionv2.onnx`)

The model files are:
Expand Down

0 comments on commit 803547b

Please sign in to comment.