ART-Net icon indicating copy to clipboard operation
ART-Net copied to clipboard

About pretrained model weights

Open xuhaozheng opened this issue 2 years ago • 0 comments

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. `

xuhaozheng avatar Jun 14 '23 14:06 xuhaozheng