From 4116cfa7bc609ca4cfd9c8e5011ea8e8f843e051 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Fri, 22 Jul 2022 17:36:44 +0800 Subject: [PATCH] Fix bugs Signed-off-by: Jael Gu --- README.md | 2 +- __init__.py | 2 ++ detectron2.py => detectron.py | 0 requirements.txt | 3 ++- 4 files changed, 5 insertions(+), 2 deletions(-) rename detectron2.py => detectron.py (100%) 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