projects
/
pysvrt.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Clean up + argument parsing + logging into a file.
[pysvrt.git]
/
test-svrt.py
diff --git
a/test-svrt.py
b/test-svrt.py
index
92fc554
..
5c16069
100755
(executable)
--- a/
test-svrt.py
+++ b/
test-svrt.py
@@
-24,16
+24,27
@@
import time
import torch
import time
import torch
+import torchvision
from torch import optim
from torch import FloatTensor as Tensor
from torch.autograd import Variable
from torch import nn
from torch.nn import functional as fn
from torch import optim
from torch import FloatTensor as Tensor
from torch.autograd import Variable
from torch import nn
from torch.nn import functional as fn
+
from torchvision import datasets, transforms, utils
from torchvision import datasets, transforms, utils
-from _ext import svrt
+import svrt
+
+labels = torch.LongTensor(12).zero_()
+labels.narrow(0, 0, labels.size(0)//2).fill_(1)
+
+x = svrt.generate_vignettes(4, labels)
+
+x = x.view(x.size(0), 1, x.size(1), x.size(2))
+
+x.div_(255)
-t
rain_set = svrt.generate_vignettes(12, 1234
)
+t
orchvision.utils.save_image(x, 'example.png'
)
-print(
str(type(train_set)), train_set.size()
)
+print(
'Wrote example.png'
)