constructelligence's picture
Add config.json (describes best.pt; Hub download-count query file)
ccfe353 verified
Raw History Blame Contribute Delete
839 Bytes
{
"model_type": "painting-vision-robotics-kit",
"library": "pytorch",
"weights": "best.pt",
"weights_sha256": "3ef02176970ddfaae1feab77e74b4ae80af361171d0211c453c8f2ef16ecce01",
"arch": "segformer_b2",
"input_size": 768,
"classes": [
"other",
"wall_unpainted",
"wall_painted",
"wall_uncertain",
"skirting",
"switch_outlet",
"ac_unit",
"door",
"window",
"wall_obstacle"
],
"material_classes": [
"other_wall_material",
"drywall"
],
"condition_classes": [],
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"temperature": null,
"material_temperature": null,
"completion_threshold": null,
"selection_epoch": 46,
"load": "python3 predict.py IMAGE --checkpoint hf://<repo> (fetches config.json, then the weights)"
}