ART-Net
ART-Net copied to clipboard
About pretrained model weights
As said in the code, there should be some pre-train model people can use to test directly. Could you please provide the file link? or maybe it was somewhere I didn't find. `#-----------------------------------Create ART-Net Model----------------------------------------------------
Load (Read) the model which saved as yaml file.
yaml_file = open('modelSaved.yaml', 'r')
"modelSaved.yaml" is the file name and should be in the same directory of this notebook script.
loaded_model_yaml = yaml_file.read() yaml_file.close() model = model_from_yaml(loaded_model_yaml) model.summary() #If you want to see the model summary then uncomment this. Uncomment is preferable.
------------------------Load the traiend Weight to the ART-Net Model--------------------------------------- model.load_weights("FineTunedmodel.hdf5") "FineTunedmodel.hdf5" is the saved weights and should be in the same directory of this notebook script. `