AntoineH/smart_augmentation
1
0
Fork 0
mirror of https://github.com/AntoineHX/smart_augmentation.git synced 2025-05-04 12:10:45 +02:00
Code Issues Projects Releases Packages Wiki Activity

modification mineurs

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-01-13 10:59:32 -05:00
parent 7732060eb1
commit 27c1890f4c
3 changed files with 32 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

2
higher/utils.py
View file

@ -195,7 +195,7 @@ def viz_sample_data(imgs, labels, fig_name='data_sample', weight_labels=None):
plt.grid(False)
plt.imshow(sample[i,].detach().numpy(), cmap=plt.cm.binary)
label = str(labels[i].item())
if weight_labels is not None : label+= ("- p %.2f" % weight_labels[i].item())
if weight_labels is not None : label+= (" - p %.2f" % weight_labels[i].item())
plt.xlabel(label)
plt.savefig(fig_name)