diff --git a/labelme/logger.py b/labelme/logger.py index 7fc998fe07d4db681ee5a207f5ed5b1658b9d5cf..cb0756a884315cdf56a7afc558b3cc16073090a5 100644 --- a/labelme/logger.py +++ b/labelme/logger.py @@ -1,8 +1,14 @@ import datetime import logging +import os import termcolor +if os.name == "nt": # Windows + import colorama + + colorama.init() + from . import __appname__ diff --git a/setup.py b/setup.py index 672482e1d257ea8d9d8981d5919dbceb8b4649ec..6e87e73749d41b1d267703ead6bffe9cc0aebcac 100644 --- a/setup.py +++ b/setup.py @@ -1,13 +1,15 @@ from __future__ import print_function import distutils.spawn +import os import re -from setuptools import find_packages -from setuptools import setup import shlex import subprocess import sys +from setuptools import find_packages +from setuptools import setup + def get_version(): filename = "labelme/__init__.py" @@ -75,6 +77,9 @@ def get_install_requires(): QT_BINDING = "pyqt5" del QT_BINDING + if os.name == "nt": # Windows + install_requires.append("colorama") + return install_requires