Skip to content
Snippets Groups Projects
Commit 05fea191 authored by Renaud's avatar Renaud
Browse files

Fix

parent 3765425c
No related branches found
No related tags found
No related merge requests found
"""
Outils permettant la création de l'application finale
@author: Groupe 7
Preuve de concept d'une solution numérique permettant d'identifier tout implant
ou prothèse à partir des informations présentes sur son emballage.
"""
__all__ = [
......
......@@ -14,7 +14,7 @@ class OCR:
# le modèle en mémoire
# Le modèle de langage "en" a été choisi car c'est celui qui donne les
# meilleurs résultats parmi les modèles "cn", "en" et "fr".
self.ocr = self.__ocr_initialize__(use_gpu)
self.ocr = self.__ocr_initialize__(use_gpu=use_gpu)
def __ocr_initialize__(self, lang='en', use_gpu=True) -> PaddleOCR:
"""
......
from poc import *
import os
# Constantes
FONT_DIR = "data/font"
......@@ -7,6 +8,9 @@ TEST_IMAGES_DIR = "data/img"
OUTPUT_DIR = "out"
def main():
if not os.path.isdir(OUTPUT_DIR):
os.mkdir(OUTPUT_DIR)
# 1. Phase de reconnaissance optiques de caractères
ocr = OCR(use_gpu=True)
groups = ocr.recognize(TEST_IMAGES_DIR + "/biomet.jpg")
......@@ -25,8 +29,6 @@ def main():
# 3. Ajout en base de données
# partie BD : strs = [group.string for group in groups]
#######
groups2 = ocr.recognize(TEST_IMAGES_DIR + "/stryker.jpg")
ocr.save_ocr(groups2, TEST_IMAGES_DIR + "/stryker.jpg", OUTPUT_DIR + "/stryker_export_step1.jpg", FONT_DIR + "/" + FONT_FILENAME)
identification2 = Identification(groups2)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment