raven / models.py
Jakub Kwiatkowski
Add model.
e986ee1
raw
history blame
490 Bytes
import os
START_IMAGE = 12000
from tensorflow.keras.models import load_model
model = load_model("saved_model/1")
from data_utils import nload, ims, DataSetFromFolder
data = nload("/home/jkwiatkowski/all/dataset/arr/val.npy")
indexes = nload("/home/jkwiatkowski/all/dataset/arr/val_target.npy")
folders = DataSetFromFolder("/home/jkwiatkowski/all/dataset/arr/RAVEN-10000-release/RAVEN-10000", file_type="dir")
properties = DataSetFromFolder(folders[:], file_type="xml", extension="val")