diff --git a/dlclive/benchmark.py b/dlclive/benchmark.py index 524d551..4cb4fb1 100644 --- a/dlclive/benchmark.py +++ b/dlclive/benchmark.py @@ -493,7 +493,7 @@ def save_inf_times( np.std(inf_times, 1) * 1.0 / np.sqrt(np.shape(inf_times)[1]), ) - #for stat in stats: + # for stat in stats: # print("Stats:", stat) data = { diff --git a/dlclive/graph.py b/dlclive/graph.py index ac33264..0841b46 100644 --- a/dlclive/graph.py +++ b/dlclive/graph.py @@ -7,11 +7,12 @@ import tensorflow as tf -vers = (tf.__version__).split('.') -if int(vers[0])==2 or int(vers[0])==1 and int(vers[1])>12: - tf=tf.compat.v1 + +vers = (tf.__version__).split(".") +if int(vers[0]) == 2 or int(vers[0]) == 1 and int(vers[1]) > 12: + tf = tf.compat.v1 else: - tf=tf + tf = tf def read_graph(file): diff --git a/dlclive/version.py b/dlclive/version.py index 52fa95e..5af39ca 100644 --- a/dlclive/version.py +++ b/dlclive/version.py @@ -7,5 +7,5 @@ """ -__version__ = "0.0.3" +__version__ = "1.0" VERSION = __version__ diff --git a/reinstall.sh b/reinstall.sh index a511686..d0da969 100755 --- a/reinstall.sh +++ b/reinstall.sh @@ -1,3 +1,3 @@ pip uninstall deeplabcut-live python3 setup.py sdist bdist_wheel -pip install dist/deeplabcut_live-0.0.3-py3-none-any.whl +pip install dist/deeplabcut_live-1.0-py3-none-any.whl diff --git a/setup.py b/setup.py index 7ed6f25..2179ffb 100755 --- a/setup.py +++ b/setup.py @@ -37,7 +37,7 @@ setuptools.setup( name="deeplabcut-live", - version="0.0.3", + version="1.0", author="A. & M. Mathis Labs", author_email="admin@deeplabcut.org", description="Class to load exported DeepLabCut networks and perform pose estimation on single frames (from a camera feed)", @@ -51,7 +51,7 @@ include_package_data=True, classifiers=( "Programming Language :: Python :: 3", - "License :: OSI Approved :: GNU Lesser General Public License v3 (LGPLv3)", + "License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)", "Operating System :: OS Independent", ), entry_points={