" data = data.reshape((n, n) + data.shape[1:]).transpose((0, 2, 1, 3) + tuple(range(4, data.ndim + 1)))\n",
" data = data.reshape((n * data.shape[1], n * data.shape[3]) + data.shape[4:])\n",
" \n",
- " imshow(data)"
+ " plt.imshow(data)"
],
"language": "python",
"metadata": {},
"collapsed": false,
"input": [
"# index four is the center crop\n",
- "imshow(net.deprocess('data', net.blobs['data'].data[4]))"
+ "plt.imshow(net.deprocess('data', net.blobs['data'].data[4]))"
],
"language": "python",
"metadata": {},