search

Kohulan / DECIMER-Image_Transformer

DECIMER: Deep Learning for Chemical Image Recognition using Efficient-Net V2 + Transformer
MIT License
197 stars 51 forks source link

Implemented new input format as numpy array representing the image #103

Closed alexey-krasnov closed 4 months ago

alexey-krasnov commented 4 months ago

A new input format for DECIMER predict_SMILES function as numpy array representing the image allows us to connect the output from DECIMER-Image-Segmentation

raw_segments = segment_chemical_structures_from_file(file_path='path_to_image_file', expand=True, visualization=False)
for segment in raw_segments:
    smiles = predict_SMILES(image_path=segment)