You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: tensorlayer/models/vgg16.py
+3Lines changed: 3 additions & 0 deletions
Original file line number
Diff line number
Diff line change
@@ -245,6 +245,7 @@ class VGG16(VGG16Base):
245
245
Examples
246
246
---------
247
247
Classify ImageNet classes with VGG16, see `tutorial_models_vgg16.py <https://github.com/tensorlayer/tensorlayer/blob/master/example/tutorial_models_vgg16.py>__`
248
+
248
249
>>> x = tf.placeholder(tf.float32, [None, 224, 224, 3])
249
250
>>> # get the whole model
250
251
>>> vgg = tl.models.VGG16(x)
@@ -255,6 +256,7 @@ class VGG16(VGG16Base):
255
256
>>> probs = tf.nn.softmax(vgg.outputs)
256
257
257
258
Extract features with VGG16 and Train a classifier with 100 classes
259
+
258
260
>>> x = tf.placeholder(tf.float32, [None, 224, 224, 3])
0 commit comments