Skip to content

Commit 55c0078

Browse files
pjbullfacebook-github-bot
authored andcommitted
Make vis module importable, remove required torch and torchvision
Summary: In PR #3311 we made densepose a pip-installable Python package. However, we forgot to include an `__init__.py` so that the `vis` module is importable. This PR adds that `__init__.py`. It was also mentioned in that PR that there is a preference not to require torch and torchvision requirements versions: #3311 (comment) This PR also removes those requirements from the `setup.py` and lists them instead in the prerequisites in the documentation. Pull Request resolved: #3382 Reviewed By: patricklabatut Differential Revision: D30462761 Pulled By: vkhalidov fbshipit-source-id: e58985ff24cc75ff1d7905a863fa040e3be99bc3
1 parent faf6587 commit 55c0078

File tree

3 files changed

+2
-3
lines changed

3 files changed

+2
-3
lines changed

projects/DensePose/densepose/vis/__init__.py

Whitespace-only changes.

projects/DensePose/doc/GETTING_STARTED.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,8 @@ DensePose can also be installed as a Python package for integration with other s
6464

6565
The following dependencies are needed:
6666
- Python >= 3.6
67-
- [PyTorch](https://pytorch.org/get-started/locally/#start-locally)
67+
- [PyTorch](https://pytorch.org/get-started/locally/#start-locally) >= 1.7 (to match [detectron2 requirements](https://detectron2.readthedocs.io/en/latest/tutorials/install.html#requirements))
68+
- [torchvision](https://pytorch.org/vision/stable/) version [compatible with your version of PyTorch](https://github.com/pytorch/vision#installation)
6869

6970
DensePose can then be installed from this repository with:
7071

projects/DensePose/setup.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,5 @@ def get_detectron2_current_version():
3838
"detectron2@git+https://github.com/facebookresearch/detectron2.git",
3939
"opencv-python-headless>=4.5.3.56",
4040
"scipy>=1.5.4",
41-
"torch>=1.9.0",
42-
"torchvision>=0.10.0",
4341
],
4442
)

0 commit comments

Comments
 (0)