download video from github; tested updated version

Author: AlexEMG

dlclive/check_install/dog_clip.avi check_install/dog_clip.avidlclive/check_install/dog_clip.avi renamed to check_install/dog_clip.avi

@@ -9,6 +9,7 @@
 import os
 import shutil
 from dlclive import benchmark_videos
+import urllib.request
 
 
 def main():
@@ -20,6 +21,11 @@ def main():
     os.makedirs(tmp_dir, exist_ok=True)
     os.chdir(tmp_dir)
 
+    # download dog test video from github:
+    url_link = 'https://github.com/DeepLabCut/DeepLabCut-live/blob/master/check_install/dog_clip.avi?raw=True'
+    urllib.request.urlretrieve(url_link,"dog_clip.avi")
+    video_file=os.path.join(url_link,"dog_clip.avi")
+    
     # download exported dog model from DeepLabCut Model Zoo
     print("Downloading full_dog model from the DeepLabCut Model Zoo...")
     model_url = "http://deeplabcut.rowland.harvard.edu/models/DLC_Dog_resnet_50_iteration-0_shuffle-0.tar.gz"
@@ -28,7 +34,7 @@ def main():
     # run benchmark videos
     print("\n Running inference...\n")
     model_dir = "DLC_Dog_resnet_50_iteration-0_shuffle-0"
-    video_file = os.path.normpath(f"{os.path.dirname(__file__)}/dog_clip.avi")
+    print(video_file)
     benchmark_videos(model_dir, video_file, display=True, resize=0.5, pcutoff=0.25)
 
     # deleting temporary files

dlclive/check_install/check_install.py

@@ -7,5 +7,5 @@
 """
 
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 VERSION = __version__

dlclive/version.py

@@ -1,3 +1,3 @@
 pip uninstall deeplabcut-live
 python3 setup.py sdist bdist_wheel
-pip install dist/deeplabcut_live-0.0.1-py3-none-any.whl
+pip install dist/deeplabcut_live-0.0.2-py3-none-any.whl

reinstall.sh

@@ -37,7 +37,7 @@
 
 setuptools.setup(
     name="deeplabcut-live",
-    version="0.0.1",
+    version="0.0.2",
     author="A. & M. Mathis Labs",
     author_email="[email protected]",
     description="Class to load exported DeepLabCut networks and perform pose estimation on single frames (from a camera feed)",