Update DBNet model files
Browse files
README.md
CHANGED
|
@@ -10,7 +10,7 @@ pipeline_tag: object-detection
|
|
| 10 |
|
| 11 |
# DBNet for Text Detection
|
| 12 |
|
| 13 |
-
Model Hub: [
|
| 14 |
|
| 15 |
This is a DBNet model for text detection, ported to Hugging Face Transformers.
|
| 16 |
It uses a ResNet-18 backbone and Feature Pyramid Network (FPN) for multi-scale feature fusion.
|
|
@@ -24,7 +24,7 @@ from transformers import pipeline
|
|
| 24 |
# Load pipeline
|
| 25 |
ocr_pipe = pipeline(
|
| 26 |
"object-detection",
|
| 27 |
-
model="
|
| 28 |
trust_remote_code=True
|
| 29 |
)
|
| 30 |
|
|
|
|
| 10 |
|
| 11 |
# DBNet for Text Detection
|
| 12 |
|
| 13 |
+
Model Hub: [shuzi-mewtant/dbnet_res18_text_detection_v0.1](https://huggingface.co/shuzi-mewtant/dbnet_res18_text_detection_v0.1)
|
| 14 |
|
| 15 |
This is a DBNet model for text detection, ported to Hugging Face Transformers.
|
| 16 |
It uses a ResNet-18 backbone and Feature Pyramid Network (FPN) for multi-scale feature fusion.
|
|
|
|
| 24 |
# Load pipeline
|
| 25 |
ocr_pipe = pipeline(
|
| 26 |
"object-detection",
|
| 27 |
+
model="shuzi-mewtant/dbnet_res18_text_detection_v0.1",
|
| 28 |
trust_remote_code=True
|
| 29 |
)
|
| 30 |
|