Commit 3d27ca1

authored

Update VGG.py

1 parent e30ecd6 commit 3d27ca1Copy full SHA for 3d27ca1

File tree

-7

lines changed

-7

lines changed

Lines changed: 7 additions & 7 deletions

Original file line number	Diff line number	Diff line change
`@@ -29,9 +29,9 @@ def VGG16(x, n_classes, is_pretrain=True):`
`29`	`29`	`x = tools.pool('pool3', x, kernel=[1,2,2,1], stride=[1,2,2,1], is_max_pool=True)`
`30`	`30`
`31`	`31`	`x = tools.FC_layer('fc6', x, out_nodes=4096)`
`32`		`- x = tools.batch_norm(x)`
	`32`	`+ #x = tools.batch_norm(x)`
`33`	`33`	`x = tools.FC_layer('fc7', x, out_nodes=4096)`
`34`		`- x = tools.batch_norm(x)`
	`34`	`+ #x = tools.batch_norm(x)`
`35`	`35`	`x = tools.FC_layer('fc8', x, out_nodes=n_classes)`
`36`	`36`
`37`	`37`	`return x`
`@@ -80,11 +80,11 @@ def VGG16N(x, n_classes, is_pretrain=True):`
`80`	`80`
`81`	`81`
`82`	`82`	`x = tools.FC_layer('fc6', x, out_nodes=4096)`
`83`		`- with tf.name_scope('batch_norm1'):`
`84`		`- x = tools.batch_norm(x)`
	`83`	`+ #with tf.name_scope('batch_norm1'):`
	`84`	`+ #x = tools.batch_norm(x)`
`85`	`85`	`x = tools.FC_layer('fc7', x, out_nodes=4096)`
`86`		`- with tf.name_scope('batch_norm2'):`
`87`		`- x = tools.batch_norm(x)`
	`86`	`+ #with tf.name_scope('batch_norm2'):`
	`87`	`+ #x = tools.batch_norm(x)`
`88`	`88`	`x = tools.FC_layer('fc8', x, out_nodes=n_classes)`
`89`	`89`
`90`	`90`	`return x`
`@@ -99,4 +99,4 @@ def VGG16N(x, n_classes, is_pretrain=True):`
`99`	`99`
`100`	`100`
`101`	`101`
`102`		`-`
	`102`	`+`

Comments

(0)