chemistry
AI4Industry commited on
Commit
146dd97
·
verified ·
1 Parent(s): 9d7bb26

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -3
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
- `202503 moldet_yolo11[size]_960_doc.pt`
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)