def collect(path="./img_data/"):
if not path.endswith("/"):
path = path + "/"
files = os.listdir(path)
files = [
file for file in files if (not os.path.isdir(file)) and (
file.endswith("-data.jpg") or file.endswith("-label.npy"))
]
prefixes = set(
file.replace("-data.jpg", "").replace("-label.npy", "")
for file in files)
for idx, prefix in enumerate(prefixes):
try:
img_file = cv2.imread(path + prefix + "-data.jpg")
img_label = np.load(path + prefix + '-label.npy')
np_image_data = np.asarray(img_file)
if np.sum(img_label > 8) > 0:
continue
print('processing {} ({}/{})'.format(prefix, idx, len(prefixes)))
tag = Tagging(np_image_data)
tag.tag(np_image_data, prefix)
img_label = np.load(path + prefix + '-label.npy')
print(img_label)
except FileExistsError as e:
print(e)
print("Loading error with prefix =", prefix, ". Skipped.")
continue
