projects
/
flatland.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[flatland.git]
/
test.py
diff --git
a/test.py
b/test.py
index
bf51360
..
c6b6c48
100755
(executable)
--- a/
test.py
+++ b/
test.py
@@
-4,6
+4,8
@@
import torch
import torchvision
from torchvision import datasets
import torchvision
from torchvision import datasets
+from _ext import flatland
+
######################################################################
def sequences_to_image(x):
######################################################################
def sequences_to_image(x):
@@
-40,10
+42,6
@@
def sequences_to_image(x):
######################################################################
######################################################################
-from _ext import mylib
-
-x = torch.ByteTensor()
-
-mylib.generate_sequence(10, x)
+x = flatland.generate_sequence(5, 3, 128, 96)
sequences_to_image(x).save('sequences.png')
sequences_to_image(x).save('sequences.png')