Przeglądaj źródła

Fix labelme_on_docker

Kentaro Wada 7 lat temu
rodzic
commit
f77477be92
1 zmienionych plików z 3 dodań i 1 usunięć
  1. 3 1
      labelme/cli/on_docker.py

+ 3 - 1
labelme/cli/on_docker.py

@@ -17,6 +17,8 @@ def get_ip():
     elif dist == 'Darwin':
         cmd = 'ifconfig en0'
         output = subprocess.check_output(shlex.split(cmd))
+        if str != bytes:  # Python3
+            output = output.decode('utf-8')
         for row in output.splitlines():
             cols = row.strip().split(' ')
             if cols[0] == 'inet':
@@ -46,7 +48,7 @@ def labelme_on_docker(image_file, out_file):
         ' -v {1}:{2}' \
         ' -v {3}:{4}' \
         ' -w /home/developer' \
-        ' labelme' \
+        ' wkentaro/labelme' \
         ' labelme {2} -O {4}'
     cmd = cmd.format(
         ip,