diff --git a/Season1/4-6/load.py b/Season1/4-6/load.py
index 0a67ffe..708c1be 100644
--- a/Season1/4-6/load.py
+++ b/Season1/4-6/load.py
@@ -69,11 +69,14 @@ def distribution(labels, name):
 
 def inspect(dataset, labels, i):
 	# 显示图片看看
+	print(labels[i])
+	'''
 	if dataset.shape[3] == 1:
 		shape = dataset.shape
 		dataset = dataset.reshape(shape[0], shape[1], shape[2])
-	print(labels[i])
 	plt.imshow(dataset[i])
+	'''#可以改为以下	
+	plt.imshow(dataset[i].squeeze())
 	plt.show()
 
 
</div><div class="naked_ctrl">
<form action="/index.cgi/contrast" method="get" name="gate">
<p><a href="http://altstyle.alfasado.net">AltStyle</a> によって変換されたページ <a href="https://patch-diff.githubusercontent.com/raw/CreatCodeBuild/TensorFlow-and-DeepLearning-Tutorial/pull/22.diff">(-&gt;オリジナル)</a>
/ <label>アドレス: <input type="text" name="naked_post_url" value="https://patch-diff.githubusercontent.com/raw/CreatCodeBuild/TensorFlow-and-DeepLearning-Tutorial/pull/22.diff" size="22" /></label> <label>モード: <select name="naked_post_mode">
<option value="default">デフォルト</option>
<option value="speech">音声ブラウザ</option>
<option value="ruby">ルビ付き</option>
<option value="contrast" selected="selected">配色反転</option>
<option value="larger-text">文字拡大</option>
<option value="mobile">モバイル</option>
</select>
<input type="submit" value="表示" />
</p>
</form>
</div>