Commit 4720935

authored

Update yolo_tf.py

1 parent 0d047f4 commit 4720935Copy full SHA for 4720935

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -131,7 +131,7 @@ def _conv_layer(self, x, id, num_filters, filter_size, stride):`
`131`	`131`	`weight = tf.Variable(tf.truncated_normal([filter_size, filter_size,`
`132`	`132`	`in_channels, num_filters], stddev=0.1))`
`133`	`133`	`bias = tf.Variable(tf.zeros([num_filters,]))`
`134`		`- # padding, note: not using padding="VALID"`
	`134`	`+ # padding, note: not using padding="SAME"`
`135`	`135`	`pad_size = filter_size // 2`
`136`	`136`	`pad_mat = np.array([[0, 0], [pad_size, pad_size], [pad_size, pad_size], [0, 0]])`
`137`	`137`	`x_pad = tf.pad(x, pad_mat)`

Comments

(0)