danila 1.0.0

This is the module for detecting and classifying text on rama pictures

To install project made

git clone
pip install -r requirements.txt

All user methods are in

Danila class

makes object and loads all neuro models

danila = Danila()

returns string - class of rama using CNN network

def rama_classify(self, image_path)

returns image_path in danilav1 root with drawn rectangle and text - rama and its class

def rama_detect(self, image_path)

returns image_path in danilav1 root with cut_rama

def rama_cut(self, image_path)

returns image-path of cut rama with drawn text areas

def text_detect_cut(self, image_path)

returns image-path of image with drawn text areas

def text_detect_cut(self, image_path)

scripts illustrates methods using

demo_1
demo_2
demo_3
demo_4
demo_5

to start work you should

add directory yolo
cd yolo/
git clone https://github.com/ultralytics/yolov5.git

you should paste directory models just in root

https://disk.yandex.ru/client/disk/%D0%9A%D0%BE%D0%BC%D0%BF%D1%8C%D1%8E%D1%82%D0%B5%D1%80%20NUFDR0019114/%D0%A0%D0%90%D0%9C%D0%90