qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-08 11:49:55 +02:00
Code Issues Projects Releases Wiki Activity

use pathlib to determine f_name

Browse source
This commit is contained in:
Konstantin Baierer 2021-01-27 11:51:23 +01:00
parent 85d631aadf
commit 2bef2d4c04
1 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
sbb_newspapers_org_image/eynollah.py
View file

@ -9,6 +9,7 @@ import random
import sys
import time
import warnings
from pathlib import Path
from multiprocessing import Process, Queue, cpu_count
from sys import getsizeof
@ -148,12 +149,8 @@ class eynollah:
self.headers_off = headers_off
self.dir_of_deskewed = dir_of_deskewed
self.dir_of_all = dir_of_all
if self.f_name is None:
try:
self.f_name = image_dir.split("/")[len(image_dir.split("/")) - 1]
self.f_name = self.f_name.split(".")[0]
except:
self.f_name = self.f_name.split(".")[0]
if not self.f_name:
self.f_name = Path(Path(image_dir).name).stem
self.dir_models = dir_models
self.kernel = np.ones((5, 5), np.uint8)