update smp

jeffreyking23 · May 25, 2022 · 18d2775 · 18d2775
1 parent ac01a9e
commit 18d2775
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pyskl/smp.py b/pyskl/smp.py
@@ -127,3 +127,11 @@ def gen_bash(cfgs, num_gpus, gpus_per_task=1):
             )
         timestamp = time.strftime('%m%d%H%M%S', time.localtime())
         mwlines(cmds, f'train_{timestamp}_{i}.sh')
+
+def h2r(value):
+    value = value.lstrip('#')
+    lv = len(value)
+    return tuple(int(value[i:i + lv // 3], 16) for i in range(0, lv, lv // 3))
+
+def r2h(rgb):
+    return '#%02x%02x%02x' % rgb