Проверил метки на символы не из словаря.
Some checks are pending
Build/Publish Develop Docs / deploy (push) Waiting to run
Some checks are pending
Build/Publish Develop Docs / deploy (push) Waiting to run
This commit is contained in:
@@ -22,14 +22,23 @@ def check_images(directory_path: str):
|
|||||||
except Exception:
|
except Exception:
|
||||||
print(file)
|
print(file)
|
||||||
|
|
||||||
def check_labels(dir_path: str) -> int:
|
def check_labels(dir_path: str):
|
||||||
for filename in os.listdir(dir_path):
|
for filename in os.listdir(dir_path):
|
||||||
name, extension = filename.split(".")
|
name, extension = filename.split(".")
|
||||||
if len(name) != 17:
|
if len(name) != 17:
|
||||||
print(filename)
|
print(filename)
|
||||||
|
|
||||||
|
def check_symbols(dir_path: str):
|
||||||
|
with open(os.path.join(dir_path, "dict.txt"), "r") as dict_file:
|
||||||
|
dict_content = dict_file.readlines()
|
||||||
|
dict_chars = set(char.strip() for char in dict_content)
|
||||||
|
for filename in os.listdir(os.path.join(dir_path, "images")):
|
||||||
|
label, extension = filename.split(".")
|
||||||
|
if any([char not in dict_chars for char in label]):
|
||||||
|
print(filename)
|
||||||
|
|
||||||
# rename_files("train_data/images/")
|
# rename_files("train_data/images/")
|
||||||
# check_images("train_data/images/")
|
# check_images("train_data/images/")
|
||||||
check_labels("train_data/images/")
|
# check_labels("train_data/images/")
|
||||||
|
check_symbols("train_data/")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user