{ "model_type": "UNetTimmWithClassification", "input_channels": 3, "num_classes": 2, "image_size": [224, 224], "framework": "pytorch", "author": "garyzgao", "description": "U-Net model with segmentation functionality for archaeological sherd classification" }