Jiahao123/tongue_segmentation_attenunet

Attenunet model to segment tongue images.

To load model from Hugging Face:

repo_id = "Jiahao123/tongue_segmentation_attenunet"

# Download files from Hugging Face
json_path = hf_hub_download(repo_id=repo_id, filename="attenunet_architecture.json")
weights_path = hf_hub_download(repo_id=repo_id, filename="attenunet_28.weights.h5")

# Load architecture
with open(json_path, "r") as json_file:
    model_json = json_file.read()
test_model = model_from_json(model_json)

# Load weights
test_model.load_weights(weights_path)