Skip to content

Commit 1c375e9

Browse files
committed
fix check_install
1 parent c7ede4c commit 1c375e9

File tree

4 files changed

+4
-8
lines changed

4 files changed

+4
-8
lines changed

MANIFEST.in

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
include dlclive/check_install/*

check_install/check_install.py renamed to dlclive/check_install/check_install.py

+1-7
Original file line numberDiff line numberDiff line change
@@ -25,16 +25,10 @@ def main():
2525
model_url = "http://deeplabcut.rowland.harvard.edu/models/DLC_Dog_resnet_50_iteration-0_shuffle-0.tar.gz"
2626
os.system(f"curl {model_url} | tar xvz")
2727

28-
# download dog video clip from github
29-
print("\nDownloading dog video clip...\n")
30-
# video_url = "https://github.com/DeepLabCut/DeepLabCut-live/raw/master/check_install/dog_clip.avi"
31-
video_url = '"https://docs.google.com/uc?export=download&id=1W_5AOl1SewXR2q5QC1K5Chm71I9LAmld"'
32-
os.system(f"curl -L {video_url} -o dog_clip.avi")
33-
3428
# run benchmark videos
3529
print("\n Running inference...\n")
3630
model_dir = "DLC_Dog_resnet_50_iteration-0_shuffle-0"
37-
video_file = "dog_clip.avi"
31+
video_file = os.path.normpath(f"{os.path.dirname(__file__)}/dog_clip.avi")
3832
benchmark_videos(model_dir, video_file, display=True, resize=0.5, pcutoff=0.25)
3933

4034
# deleting temporary files
File renamed without changes.

setup.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
python_requires=">=3.5, <3.8",
4848
install_requires=install_requires,
4949
packages=setuptools.find_packages(),
50+
package_data={'dlclive': ['check_install/*']},
5051
include_package_data=True,
5152
classifiers=(
5253
"Programming Language :: Python :: 3",
@@ -55,7 +56,7 @@
5556
),
5657
entry_points={
5758
"console_scripts": [
58-
"dlc-live-test=check_install.check_install:main",
59+
"dlc-live-test=dlclive.check_install.check_install:main",
5960
"dlc-live-bench=dlclive.bench:main",
6061
"dlc-live-benchmark=dlclive.benchmark:main",
6162
]

0 commit comments

Comments
 (0)