mirror of
https://github.com/igv/FSRCNN-TensorFlow.git
synced 2026-08-17 16:51:49 +08:00
Script for sorting deconvolution filter weights
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user