Script for sorting deconvolution filter weights

This commit is contained in:
igv
2017-09-08 15:41:49 +03:00
parent 7e828beb9a
commit 286e6cfc3b
+39
View File
@@ -0,0 +1,39 @@
import sys
def main():
scale = 2
radius = [3, 5, 7][scale-2]
d = 64 #size of the feature layer
if len(sys.argv) == 2:
fname=sys.argv[1]
with open(fname) as f:
content = f.readlines()
content = [x.strip() for x in content]
xy = []
for i in range(0, scale):
for j in range(0, scale):
xy.append([j, i])
xy = list(reversed(xy))
m = []
for i in range(0, len(xy)):
xi, yi = xy[i]
for x in range(xi, radius*2+1, scale):
for y in range(yi, radius*2+1, scale):
m.append(y + x*(radius*2+1))
#print(m)
content = list(reversed(content))
sort = [content[m[l]].strip(",") for l in range(0, len(m))]
with open('sorted.txt', 'w') as file:
for i in range(0, d, 4):
for l in range(0, len(sort)):
m=sort[l].strip(",").split(",")
file.write(",".join(m[i:i+4])+"\n")
file.write("\n")
else:
print("Missing argument: You must specify a file name that contains the deconvolution weights")
return
if __name__ == '__main__':
main()