Skip to content

About pretrained model weights #12

@xuhaozheng

Description

@xuhaozheng

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions