Skip to content

Commit 5740ffc

Browse files
committed
Setup fix for installing detectron2
1 parent 688d10f commit 5740ffc

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

.github/workflows/ci.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ jobs:
2121
run: |
2222
python -m pip install --upgrade pip
2323
pip install .
24+
pip install .[DL]
2425
2526
- name: Lint with flake8
2627
run: |

setup.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,14 @@
2626
"pyyaml>=5.1",
2727
"torch==1.4",
2828
"torchvision==0.5",
29-
"detectron2 @ git+https://github.com/facebookresearch/detectron2.git@v0.1.3#egg=detectron2"
3029
],
3130
extras_require={
3231
"GCV": ['google-cloud-vision'],
33-
"Tesseract": ['pytesseract']
32+
"Tesseract": ['pytesseract'],
33+
"DL": [
34+
"pycocotools",
35+
"detectron2 @ git+https://github.com/facebookresearch/detectron2.git@v0.1.3#egg=detectron2",
36+
]
3437
},
3538
include_package_data=True
3639
)

0 commit comments

Comments
 (0)