Move to xautodl

2021-05-18 14:08:00 +00:00
parent 98fadf8086
commit 94a149b33f
149 changed files with 94 additions and 21 deletions
--- a/xautodl/models/shape_infers/shared_utils.py
+++ b/xautodl/models/shape_infers/shared_utils.py
@@ -0,0 +1,5 @@
+def parse_channel_info(xstring):
+    blocks = xstring.split(" ")
+    blocks = [x.split("-") for x in blocks]
+    blocks = [[int(_) for _ in x] for x in blocks]
+    return blocks