diff --git a/tensorflow2fluid/README.md b/tensorflow2fluid/README.md index 01448ca108dfe6455acd516e27506908c36b3986..e818da224198937387c0178b50232cc9c7a40fe4 100644 --- a/tensorflow2fluid/README.md +++ b/tensorflow2fluid/README.md @@ -85,6 +85,19 @@ ref_name.info|my_model.py中各tensor与原TensorFlow模型中的tensor对应关 const_\*/params_\*|转换后的模型参数文件 save_var.list|模型载入过程中的变量list +## 支持模型 +tensorflow2fluid在如下tensorflow模型上测试了模型转换前后的diff +| 模型类别| 模型 | code |最大diff | +|--------------| ------------- | --------- |-------- | +|图像分类| [VGG_16] | [code](https://github.com/tensorflow/models/blob/master/research/slim/nets/vgg.py) | 1.04E-05 | +| | [VGG_19] | [code](https://github.com/tensorflow/models/blob/master/research/slim/nets/vgg.py) | 9.07E-06 | +| | [ResNet V1 50] | [code](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v1.py) | 1.31E-06 | +| | [ResNet V1 101] | [code](https://github.com/tensorflow/models/blob/master/research/slim/nets/resnet_v1.py) | 4.74E-07 | +| | [Inception V3] | [code](https://github.com/tensorflow/models/blob/master/research/slim/nets/inception_v3.py) | 1.55E-04 | +|目标检测| [YOLO-Small] | [code](https://github.com/gliese581gg/YOLO_tensorflow) | 1.40E-06 | +| | [YOLO-V3] | [code](https://github.com/mystic123/tensorflow-yolo-v3) | 6.20E-04 | +|语义分割| [Unet] | [code](https://github.com/jakeret/tf_unet) | 4.17E-07 | + ## Link 本目录下部分代码参考了MMdnn-Tensorflow,对此表示感谢!