зеркало из 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**
|
||||
|
||||
- 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.
|
||||
|
||||
- **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,
|
||||
|
||||
![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)
|
||||
|
||||
|
|
|
@ -9,8 +9,8 @@ import tensorflow as tf
|
|||
from tensorflow.core.framework import graph_pb2
|
||||
import tfcoreml as tf_converter
|
||||
|
||||
TMP_MODEL_DIR = '/Users/kit/tmp/tfcoreml'
|
||||
TEST_IMAGE = '/Users/kit/github/MMdnn/mmdnn/conversion/examples/data/seagull.jpg'
|
||||
TMP_MODEL_DIR = 'tmp/tfcoreml'
|
||||
TEST_IMAGE = 'mmdnn/conversion/examples/data/seagull.jpg'
|
||||
|
||||
def _download_file(url):
|
||||
"""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']
|
||||
|
||||
$ 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]
|
||||
PyTorch pretrained model is saved as [./imagenet_resnet101.pth].
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче