diff --git a/README.md b/README.md index 78d9d2b..1a1742d 100644 --- a/README.md +++ b/README.md @@ -15,7 +15,7 @@ This operator uses Facebook's [Detectron2](https://github.com/facebookresearch/d ```python import towhee -towhee.glob('./towhee.jpg') \ +towhee.glob('./example.jpg') \ .image_decode() \ .object_detection.detectron2(model_name='retinanet_resnet50') \ .show() diff --git a/__init__.py b/__init__.py index 9bb167a..a635800 100644 --- a/__init__.py +++ b/__init__.py @@ -1,3 +1,5 @@ +from .detectron import Detectron2 + def detectron2(*args, **kwargs): return Detectron2(*args, **kwargs) diff --git a/detectron2.py b/detectron.py similarity index 100% rename from detectron2.py rename to detectron.py diff --git a/requirements.txt b/requirements.txt index 3b08d0f..f71aa98 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ torch >= 1.8 torchvision >= 0.9.0 -https://github.com/facebookresearch/detectron2/archive/refs/tags/v0.6.tar.gz +detectron2 +# https://github.com/facebookresearch/detectron2/archive/refs/tags/v0.6.tar.gz