Commit 0072991d authored by Sebastian N.'s avatar Sebastian N.

Updated target code

parent 9d751e4e
Pipeline #203326 failed with stages
in 18 seconds
......@@ -354,7 +354,7 @@ class CNNSupervisedTrainer_Alexnet:
attention_resized = np.resize(attention.asnumpy(), (8, 8))
ax = fig.add_subplot(max_length//3, max_length//4, l+1)
ax.set_title(dict[int(labels[l+1][0].asscalar())])
img = ax.imshow(train_images[0+batch_size*(batch_i)])
img = ax.imshow(train_images[0+test_batch_size*(batch_i)])
ax.imshow(attention_resized, cmap='gray', alpha=0.6, extent=img.get_extent())
......@@ -406,7 +406,7 @@ class CNNSupervisedTrainer_Alexnet:
attention_resized = np.resize(attention.asnumpy(), (8, 8))
ax = fig.add_subplot(max_length//3, max_length//4, l+1)
ax.set_title(dict[int(mx.nd.slice_axis(mx.nd.argmax(outputs[l+1], axis=1), axis=0, begin=0, end=1).asscalar())])
img = ax.imshow(test_images[0+batch_size*(batch_i)])
img = ax.imshow(test_images[0+test_batch_size*(batch_i)])
ax.imshow(attention_resized, cmap='gray', alpha=0.6, extent=img.get_extent())
......
......@@ -354,7 +354,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork:
attention_resized = np.resize(attention.asnumpy(), (8, 8))
ax = fig.add_subplot(max_length//3, max_length//4, l+1)
ax.set_title(dict[int(labels[l+1][0].asscalar())])
img = ax.imshow(train_images[0+batch_size*(batch_i)])
img = ax.imshow(train_images[0+test_batch_size*(batch_i)])
ax.imshow(attention_resized, cmap='gray', alpha=0.6, extent=img.get_extent())
......@@ -406,7 +406,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork:
attention_resized = np.resize(attention.asnumpy(), (8, 8))
ax = fig.add_subplot(max_length//3, max_length//4, l+1)
ax.set_title(dict[int(mx.nd.slice_axis(mx.nd.argmax(outputs[l+1], axis=1), axis=0, begin=0, end=1).asscalar())])
img = ax.imshow(test_images[0+batch_size*(batch_i)])
img = ax.imshow(test_images[0+test_batch_size*(batch_i)])
ax.imshow(attention_resized, cmap='gray', alpha=0.6, extent=img.get_extent())
......
......@@ -354,7 +354,7 @@ class CNNSupervisedTrainer_VGG16:
attention_resized = np.resize(attention.asnumpy(), (8, 8))
ax = fig.add_subplot(max_length//3, max_length//4, l+1)
ax.set_title(dict[int(labels[l+1][0].asscalar())])
img = ax.imshow(train_images[0+batch_size*(batch_i)])
img = ax.imshow(train_images[0+test_batch_size*(batch_i)])
ax.imshow(attention_resized, cmap='gray', alpha=0.6, extent=img.get_extent())
......@@ -406,7 +406,7 @@ class CNNSupervisedTrainer_VGG16:
attention_resized = np.resize(attention.asnumpy(), (8, 8))
ax = fig.add_subplot(max_length//3, max_length//4, l+1)
ax.set_title(dict[int(mx.nd.slice_axis(mx.nd.argmax(outputs[l+1], axis=1), axis=0, begin=0, end=1).asscalar())])
img = ax.imshow(test_images[0+batch_size*(batch_i)])
img = ax.imshow(test_images[0+test_batch_size*(batch_i)])
ax.imshow(attention_resized, cmap='gray', alpha=0.6, extent=img.get_extent())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment