go on

README.md

@@ -0,0 +1,3 @@
+ImageIdentification-vgg16
+
+使用keras后端和vgg16模型，识别图像。

mian.py → main.py

@@ -3,7 +3,7 @@
 '''
 @Auther :liuyuqi.gov@msn.cn
 @Time :2018/4/10 17:35
-@File :mian.py
+@File :main.py
 '''
 
 if __name__ == '__main__':

vgg16模型图像识别.ipynb

@@ -7,7 +7,8 @@
    },
    "level": 1,
    "source": [
-    "## 数据下载\n",
+    "# 数据下载\n",
+    "\n",
     "对应的模型在 'vgg16' 可以下载链接: https://pan.baidu.com/s/1qgx5LfTOen9MDlTdfGKFhw 密码: s4m5"
    ]
   },
@@ -96,7 +97,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model = VGG_16('vgg16_weights.h5')  "
+    "model = VGG_16('data/vgg16_weights.h5')  "
    ]
   },
   {
@@ -143,9 +144,9 @@
    "outputs": [],
    "source": [
     "def predict(url):\n",
-    "im = load_image(url)\n",
-    "pre = np.argmax(model.predict(im))\n",
-    "print lines[pre]"
+    "    im = load_image(url)\n",
+    "    pre = np.argmax(model.predict(im))\n",
+    "    print(lines[pre])"
    ]
   },
   {