Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,43 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
---
|
| 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 |
+
|
| 11 |
+
## general molecule detection
|
| 12 |
+
|
| 13 |
+
`moldet_yolo11[size]_640_general.pt`
|
| 14 |
+
|
| 15 |
+
* 640x640 input resolution
|
| 16 |
+
* support handwritten molecules
|
| 17 |
+
* multiscale input (inputs can be single/multiple molecular cutouts, reaction or table cutouts, or single-page PDF images)
|
| 18 |
+
|
| 19 |
+
<span style='color:gray'>For single-molecule input (used as a classification model), appropriate padding can be added to enhance the performance.</span>
|
| 20 |
+
|
| 21 |
+
Result in private testing:
|
| 22 |
+
| size | map50 | map50-95 |
|
| 23 |
+
| ---- | ----- | -------- |
|
| 24 |
+
|n | 0.9581 |0.85235 |
|
| 25 |
+
|s|0.96522|0.87038 |
|
| 26 |
+
|m|0.96857|0.87355 |
|
| 27 |
+
|l|0.98912|0.90282 |
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
## PDF molecule detection
|
| 31 |
+
|
| 32 |
+
`202503 moldet_yolo11[size]_960_doc.pt`
|
| 33 |
+
|
| 34 |
+
* 960x960 input resolution
|
| 35 |
+
* single page PDF image input
|
| 36 |
+
|
| 37 |
+
Result in private testing:
|
| 38 |
+
| size | map50 | map50-95 |
|
| 39 |
+
| ---- | ----- | -------- |
|
| 40 |
+
|n|0.98714|0.87323 |
|
| 41 |
+
|s|0.98512|0.88236 |
|
| 42 |
+
|m|0.98671|0.89169 |
|
| 43 |
+
|l|0.99125|0.90114 |
|