Skip to content

Commit 49a0c1c

Browse files
authored
setup: update opencv and tensorflow
1 parent 0c592e1 commit 49a0c1c

File tree

1 file changed

+3
-11
lines changed

1 file changed

+3
-11
lines changed

setup.py

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -25,19 +25,11 @@
2525
"tqdm",
2626
]
2727

28-
if find_spec("cv2") is None:
29-
install_requires.append("opencv-python")
30-
if find_spec("tensorflow") is None:
31-
warnings.warn(
32-
"tensorflow is not yet installed. Installing tensorflow CPU version. if you wish to use the GPU version, please run: pip install tensorflow-gpu==1.13.1"
33-
)
34-
install_requires.append("tensorflow==1.13.1")
3528
if "tegra" in platform.platform():
36-
if find_spec("pandas") is None:
37-
warnings.warn(
38-
"Not installing pandas or pytables (these can take forever on NVIDIA Jetson boards). Please install these packages if you want to use the benchmark_videos function to save poses from a video."
39-
)
29+
warnings.warn("Not installing the following packages:\nopencv-python\ntensorflow\npandas\ntables\nPlease follow instructions on github to install opencv and tensorflow. If you want to use the benchmark_videos function to save poses from a video, then please install pandas and tables (pip install pandas tables)")
4030
else:
31+
install_requires.append("opencv-python")
32+
install_requires.append("tensorflow==1.13.1")
4133
install_requires.append("pandas")
4234
install_requires.append("tables")
4335

0 commit comments

Comments
 (0)