Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,11 @@ from rdkit import Chem
|
|
| 8 |
from rdkit.Chem import Draw
|
| 9 |
from utils import bbox_to_graph_with_charge
|
| 10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
def preprocess_image(image):
|
| 12 |
"""
|
| 13 |
预处理输入图片
|
|
|
|
| 8 |
from rdkit.Chem import Draw
|
| 9 |
from utils import bbox_to_graph_with_charge
|
| 10 |
|
| 11 |
+
bond_labels = [13,14,15,16,17]
|
| 12 |
+
idx_to_labels = {0:'other',1:'C',2:'O',3:'N',4:'Cl',5:'Br',6:'S',7:'F',8:'B',
|
| 13 |
+
9:'I',10:'P',11:'*',12:'Si',13:'NONE',14:'BEGINWEDGE',15:'BEGINDASH',
|
| 14 |
+
16:'=',17:'#',18:'-4',19:'-2',20:'-1',21:'1',22:'+2',} #NONE is single ?
|
| 15 |
+
|
| 16 |
def preprocess_image(image):
|
| 17 |
"""
|
| 18 |
预处理输入图片
|