Update README.md
Browse files
README.md
CHANGED
|
@@ -4,7 +4,7 @@ license: mit
|
|
| 4 |
|
| 5 |
# Molecule Detection YOLO in MolParser
|
| 6 |
|
| 7 |
-
From paper: *MolParser: End-to-end Visual Recognition of Molecule Structures in the Wild*
|
| 8 |
|
| 9 |
We provide several [ultralytics YOLO11]((https://github.com/ultralytics/ultralytics)) weights for molecule detection with different size & input resolution.
|
| 10 |
|
|
@@ -26,10 +26,16 @@ Result in private testing:
|
|
| 26 |
|m|0.96857|0.87355 |
|
| 27 |
|l|0.98912|0.90282 |
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
## PDF molecule detection
|
| 31 |
|
| 32 |
-
`
|
| 33 |
|
| 34 |
* 960x960 input resolution
|
| 35 |
* single page PDF image input
|
|
@@ -40,4 +46,10 @@ Result in private testing:
|
|
| 40 |
|n|0.98714|0.87323 |
|
| 41 |
|s|0.98512|0.88236 |
|
| 42 |
|m|0.98671|0.89169 |
|
| 43 |
-
|l|0.99125|0.90114 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
# Molecule Detection YOLO in MolParser
|
| 6 |
|
| 7 |
+
From paper: "*MolParser: End-to-end Visual Recognition of Molecule Structures in the Wild*" (ICCV2025 under review)
|
| 8 |
|
| 9 |
We provide several [ultralytics YOLO11]((https://github.com/ultralytics/ultralytics)) weights for molecule detection with different size & input resolution.
|
| 10 |
|
|
|
|
| 26 |
|m|0.96857|0.87355 |
|
| 27 |
|l|0.98912|0.90282 |
|
| 28 |
|
| 29 |
+
usage:
|
| 30 |
+
```python
|
| 31 |
+
from ultralytics import YOLO
|
| 32 |
+
model = YOLO("moldet_yolo11l_640_general.pt")
|
| 33 |
+
model.predict("path/to/image.png", save=True, imgsz=640, conf=0.5)
|
| 34 |
+
```
|
| 35 |
|
| 36 |
## PDF molecule detection
|
| 37 |
|
| 38 |
+
`moldet_yolo11[size]_960_doc.pt`
|
| 39 |
|
| 40 |
* 960x960 input resolution
|
| 41 |
* single page PDF image input
|
|
|
|
| 46 |
|n|0.98714|0.87323 |
|
| 47 |
|s|0.98512|0.88236 |
|
| 48 |
|m|0.98671|0.89169 |
|
| 49 |
+
|l|0.99125|0.90114 |
|
| 50 |
+
|
| 51 |
+
usage:
|
| 52 |
+
```python
|
| 53 |
+
from ultralytics import YOLO
|
| 54 |
+
model = YOLO("moldet_yolo11l_960_doc.pt")
|
| 55 |
+
model.predict("path/to/pdf_page_image.png", save=True, imgsz=960, conf=0.5)
|