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

Commentaires + rangement

Browse source
This commit is contained in:
Harle, Antoine (Contracteur) 2020-01-24 11:50:30 -05:00
parent da711d17cd
commit 2e09f07f52
4 changed files with 336 additions and 279 deletions
Download patch file Download diff file Expand all files Collapse all files

4
higher/utils.py
View file

@ -121,7 +121,7 @@ def viz_sample_data(imgs, labels, fig_name='data_sample', weight_labels=None):
plt.figure(figsize=(10,10))
for i in range(25):
plt.subplot(5,5,i+1)
plt.subplot(5,5,i+1) #Trop de figure cree ?
plt.xticks([])
plt.yticks([])
plt.grid(False)
@ -132,7 +132,7 @@ def viz_sample_data(imgs, labels, fig_name='data_sample', weight_labels=None):
plt.savefig(fig_name)
print("Sample saved :", fig_name)
plt.close()
plt.close('all')
def print_torch_mem(add_info=''):