зеркало из https://github.com/microsoft/MMdnn.git
Normalize line ending and fix absolute paths. (#855)
This commit is contained in:
Родитель
6e9b8a722a
Коммит
8836a43100
|
@ -19,7 +19,7 @@ Major features include:
|
||||||
|
|
||||||
- **Model Search & Visualization**
|
- **Model Search & Visualization**
|
||||||
|
|
||||||
- We provide a [model collection](https://github.com/Microsoft/MMdnn/blob/master/mmdnn/models/README.md) to help you find some popular models.
|
- We provide a [model collection](mmdnn/models/README.md) to help you find some popular models.
|
||||||
- We provide a <a href="#visualization">model visualizer</a> to display the network architecture more intuitively.
|
- We provide a <a href="#visualization">model visualizer</a> to display the network architecture more intuitively.
|
||||||
|
|
||||||
- **Model Deployment**
|
- **Model Deployment**
|
||||||
|
|
|
@ -37,7 +37,7 @@ TensorBoard 1.5.1 at http://kit-station:6006 (Press CTRL+C to quit)
|
||||||
|
|
||||||
Then you can open URL above to find the output node of your model,
|
Then you can open URL above to find the output node of your model,
|
||||||
|
|
||||||
![TensorBoard](https://github.com/Microsoft/MMdnn/blob/master/docs/tensorboard.png)
|
![TensorBoard](tensorboard.png)
|
||||||
|
|
||||||
like the squeeze node named *MMdnn_Output* we set up in our tensorflow model extractor. Detail information is in [TensorFlow README](https://github.com/Microsoft/MMdnn/blob/master/mmdnn/conversion/tensorflow/README.md)
|
like the squeeze node named *MMdnn_Output* we set up in our tensorflow model extractor. Detail information is in [TensorFlow README](https://github.com/Microsoft/MMdnn/blob/master/mmdnn/conversion/tensorflow/README.md)
|
||||||
|
|
||||||
|
|
|
@ -9,8 +9,8 @@ import tensorflow as tf
|
||||||
from tensorflow.core.framework import graph_pb2
|
from tensorflow.core.framework import graph_pb2
|
||||||
import tfcoreml as tf_converter
|
import tfcoreml as tf_converter
|
||||||
|
|
||||||
TMP_MODEL_DIR = '/Users/kit/tmp/tfcoreml'
|
TMP_MODEL_DIR = 'tmp/tfcoreml'
|
||||||
TEST_IMAGE = '/Users/kit/github/MMdnn/mmdnn/conversion/examples/data/seagull.jpg'
|
TEST_IMAGE = 'mmdnn/conversion/examples/data/seagull.jpg'
|
||||||
|
|
||||||
def _download_file(url):
|
def _download_file(url):
|
||||||
"""Download the file.
|
"""Download the file.
|
||||||
|
|
|
@ -30,7 +30,7 @@ $ mmdownload -f pytorch -h
|
||||||
Support frameworks: ['alexnet', 'densenet121', 'densenet161', 'densenet169', 'densenet201', 'inception_v3', 'resnet101', 'resnet152', 'resnet18', 'resnet34', 'resnet50', 'vgg11', 'vgg11_bn', 'vgg13', 'vgg13_bn', 'vgg16', 'vgg16_bn', 'vgg19', 'vgg19_bn']
|
Support frameworks: ['alexnet', 'densenet121', 'densenet161', 'densenet169', 'densenet201', 'inception_v3', 'resnet101', 'resnet152', 'resnet18', 'resnet34', 'resnet50', 'vgg11', 'vgg11_bn', 'vgg13', 'vgg13_bn', 'vgg16', 'vgg16_bn', 'vgg19', 'vgg19_bn']
|
||||||
|
|
||||||
$ mmdownload -f pytorch -n resnet101 -o ./
|
$ mmdownload -f pytorch -n resnet101 -o ./
|
||||||
Downloading: "https://download.pytorch.org/models/resnet101-5d3b4d8f.pth" to /home/ruzhang/.torch/models/resnet101-5d3b4d8f.pth
|
Downloading: "https://download.pytorch.org/models/resnet101-5d3b4d8f.pth" to /my/home/.torch/models/resnet101-5d3b4d8f.pth
|
||||||
███████████████████| 102502400/102502400 [00:06<00:00, 15858546.50it/s]
|
███████████████████| 102502400/102502400 [00:06<00:00, 15858546.50it/s]
|
||||||
PyTorch pretrained model is saved as [./imagenet_resnet101.pth].
|
PyTorch pretrained model is saved as [./imagenet_resnet101.pth].
|
||||||
|
|
||||||
|
|
Загрузка…
Ссылка в новой задаче