Commit dcfc8b2

authored

Merge pull request #22 from twiceyuan/master

Update load.py

2 parents 338efd6 + caf148b commit dcfc8b2Copy full SHA for dcfc8b2

File tree

-1

lines changed

-1

lines changed

Lines changed: 4 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -69,11 +69,14 @@ def distribution(labels, name):`
`69`	`69`
`70`	`70`	`def inspect(dataset, labels, i):`
`71`	`71`	`# 显示图片看看`
	`72`	`+ print(labels[i])`
	`73`	`+ '''`
`72`	`74`	`if dataset.shape[3] == 1:`
`73`	`75`	`shape = dataset.shape`
`74`	`76`	`dataset = dataset.reshape(shape[0], shape[1], shape[2])`
`75`		`- print(labels[i])`
`76`	`77`	`plt.imshow(dataset[i])`
	`78`	`+ '''#可以改为以下`
	`79`	`+ plt.imshow(dataset[i].squeeze())`
`77`	`80`	`plt.show()`
`78`	`81`
`79`	`82`

Comments

(0)