From 82e430896594a9be0a97290b01f0acd3d568b36b Mon Sep 17 00:00:00 2001
From: Paul Best <paul.best@lis-lab.fr>
Date: Thu, 26 Jan 2023 15:17:31 +0100
Subject: [PATCH] stdc2txt.py
---
stdc2txt.py | 26 ++++++++++++++++++++++++++
1 file changed, 26 insertions(+)
create mode 100755 stdc2txt.py
diff --git a/stdc2txt.py b/stdc2txt.py
new file mode 100755
index 0000000..bb54fe4
--- /dev/null
+++ b/stdc2txt.py
@@ -0,0 +1,26 @@
+import os
+import argparse
+import torch
+import numpy as np
+
+parser = argparse.ArgumentParser()
+parser.add_argument("stdc")
+args = parser.parse_args()
+
+os.system('rm output_stdc.txt')
+
+f = open('output_stdc.txt', 'a')
+m = torch.load(args.stdc)
+
+for k in m.keys():
+ m[k] = m[k].squeeze().cpu()
+ f.write(k+'\n')
+ if len(m[k].shape) == 0:
+ f.write(str(m[k].item())+'\n')
+ elif len(m[k].shape) == 1:
+ f.write(",".join(m[k].numpy().astype(str))+'\n')
+ else :
+ for l in m[k].squeeze():
+ f.write(",".join(l.numpy().astype(str))+'\n')
+ f.write('\n')
+
--
GitLab