def load_data():
print("Loading data...")
X, y = np.asarray([], dtype="float32"), np.asarray([], dtype="int32")
cats = load_classes()
for cat in cats:
delete_broken_images(cat.name)
imgs = load_images_from_path(cat.name)
X = np.concatenate((X,imgs))
y = np.concatenate((y, [cats.index(cat)]*len(imgs)))
print("Loaded {0} images for category {1}".format(len(imgs), cat.name))
print("Loaded data.")
return X,y
def load_g_data():
print("Loading data...")
X, y = np.asarray([], dtype="float32"), np.asarray([], dtype="int32")
cats = load_classes()
for cat in cats:
try:
path = "g_" + cat.name
delete_broken_images(path)
imgs = load_images_from_path(path)
X = np.concatenate((X,imgs))
y = np.concatenate((y, [cats.index(cat)]*len(imgs)))
except:
print("F**k you")
print("Loaded data.")
return X,y
def fix_data(): cats = load_classes() for cat in cats: delete_broken_images(cat.name)