projects
/
mygpt.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
52c6bd9
)
OCDC
author
Francois Fleuret
<francois@fleuret.org>
Fri, 29 Jul 2022 04:04:57 +0000
(06:04 +0200)
committer
Francois Fleuret
<francois@fleuret.org>
Fri, 29 Jul 2022 04:04:57 +0000
(06:04 +0200)
picoclvr.py
patch
|
blob
|
history
diff --git
a/picoclvr.py
b/picoclvr.py
index
c455072
..
f097eb0
100755
(executable)
--- a/
picoclvr.py
+++ b/
picoclvr.py
@@
-163,6
+163,7
@@
def descr2properties(descr, height, width):
seen = {}
if len(d) != height * width: return []
seen = {}
if len(d) != height * width: return []
+
for k, x in enumerate(d):
if x != color_names[0]:
if x in color_tokens:
for k, x in enumerate(d):
if x != color_names[0]:
if x in color_tokens:
@@
-171,9
+172,10
@@
def descr2properties(descr, height, width):
return []
seen[x] = (color_id[x], k // width, k % width)
return []
seen[x] = (color_id[x], k // width, k % width)
- square_c = torch.tensor( [ x[0] for x in seen.values() ] )
- square_i = torch.tensor( [ x[1] for x in seen.values() ] )
- square_j = torch.tensor( [ x[2] for x in seen.values() ] )
+ square_infos = zip(*seen.values())
+ square_c = torch.tensor(square_infos[0])
+ square_i = torch.tensor(square_infos[1])
+ square_j = torch.tensor(square_infos[2])
s = all_properties(height, width, len(seen), square_i, square_j, square_c)
s = all_properties(height, width, len(seen), square_i, square_j, square_c)