DPSeg / released_weights.json
Lemma456's picture
Publish DPSeg ablation checkpoints and verified file manifest
9ea4aef verified
Raw History Blame Contribute Delete
5.14 kB
{
"schema": "dpseg-released-weights-v1",
"checkpoint_format": "Original PyTorch checkpoint bytes; torch.load(..., weights_only=True).",
"scope": "Only three selected checkpoints required by the five progressive ablations S0-S4; the point checkpoint includes plain and depth heads. Historical experiments, intermediate epochs, and optimizer states are excluded.",
"stages": [
{
"id": "S0",
"name": "Mask2Former fine-tune",
"weights": [
"rgb_e15"
],
"point_head": null,
"tiled": false,
"probability_depth": false
},
{
"id": "S1",
"name": "+ RGB-D 融合",
"weights": [
"rgbd_e15"
],
"point_head": null,
"tiled": false,
"probability_depth": false
},
{
"id": "S2",
"name": "+ 深度点细化",
"weights": [
"rgbd_e15",
"points_base15_e06"
],
"point_head": "depth",
"tiled": false,
"probability_depth": false
},
{
"id": "S3",
"name": "+ 切片推理",
"weights": [
"rgbd_e15",
"points_base15_e06"
],
"point_head": "depth",
"tiled": true,
"probability_depth": false
},
{
"id": "S4",
"name": "+ 概率图深度细化",
"weights": [
"rgbd_e15",
"points_base15_e06"
],
"point_head": "depth",
"tiled": true,
"probability_depth": true
}
],
"weights": [
{
"id": "rgb_e15",
"group": "main",
"description": "Mask2Former RGB,1024,15轮;S0",
"filename": "dpseg_rgb_swin_t_1024_e15.pt",
"url": "https://huggingface.co/Lemma456/DPSeg/resolve/weights-v1/dpseg_rgb_swin_t_1024_e15.pt",
"bytes": 190104323,
"sha256": "b8e7f01ad167a5474943d81dc2633d4557d15d127d02cec0abc2df1c775bf3ff",
"epoch": 15,
"state_dict_key": "model",
"tensor_sha256": "850a0f5321367a0ac0ae58158de3823d5018529fce1e72dacf06358b748b2cb0",
"protocol_sha256": "5f1dac1927e2045b002a06ac6c6f6d1b136993d5ebc26feef3601bd540140753",
"source_files_verified": 42,
"validation_metrics": {
"mask_ap": 0.668337855310945,
"mask_ap50": 0.8844866044969915,
"mask_ap75": 0.7350799335968378,
"mask_ap_small": 0.3431467071437425,
"mask_ap_medium": 0.7002932027973691,
"mask_ap_large": 0.8421113602192869,
"mask_ar_1": 0.03288841067269582,
"mask_ar_10": 0.2835115293989458,
"mask_ar_100": 0.7113373557973877,
"mask_ar_small": 0.4085371397399764,
"mask_ar_medium": 0.7420744019881951,
"mask_ar_large": 0.8706570410307961
}
},
{
"id": "rgbd_e15",
"group": "main",
"description": "DA-Fusion RGB-D,1024,15轮;S1–S4",
"filename": "dpseg_rgbd_swin_t_1024_e15.pt",
"url": "https://huggingface.co/Lemma456/DPSeg/resolve/weights-v1/dpseg_rgbd_swin_t_1024_e15.pt",
"bytes": 351068109,
"sha256": "ef52456da6bb47e4f38c7cfb26e5eaf88c02895bf674487febd04d24d0dc2059",
"epoch": 15,
"state_dict_key": "model",
"tensor_sha256": "8c1549a03330f4448146bdb0986e7fa7895c8ccaf15d1e903fe6106a487dc314",
"protocol_sha256": "e3ebd11c7d3699b5156805eac8e26f6e9944b82eb68741009ff35dea5701d650",
"source_files_verified": 37,
"validation_metrics": {
"mask_ap": 0.8148707314438498,
"mask_ap50": 0.9422293177974336,
"mask_ap75": 0.8757028109697857,
"mask_ap_small": 0.5037415129909696,
"mask_ap_medium": 0.8422145347059979,
"mask_ap_large": 0.9678109735428819,
"mask_ar_1": 0.034176870377684854,
"mask_ar_10": 0.3126326255431008,
"mask_ar_100": 0.8337614578935183,
"mask_ar_small": 0.5611328302572962,
"mask_ar_medium": 0.862202935694315,
"mask_ar_large": 0.9735721465777208
}
},
{
"id": "points_base15_e06",
"group": "main",
"description": "第15轮基座的普通/深度点头,第6轮;S2–S4及整图六组对照",
"filename": "dpseg_points_base15_e06.pt",
"url": "https://huggingface.co/Lemma456/DPSeg/resolve/weights-v1/dpseg_points_base15_e06.pt",
"bytes": 1806733,
"sha256": "ec72065e3dd5e869d9ddd217cbddf89b2a3f9d11dd29447e57c38a9e6323ebff",
"epoch": 6,
"state_dict_key": "heads",
"tensor_sha256": "e5df8b59ab876f39d91324833c4d340d3e87ed4480703717160e1fff594ab8d7",
"protocol_sha256": "e7e1f707c4b85184273e41e3e3ebd357ee1878dc12a566b861cc9d5980519753",
"source_files_verified": 49,
"validation_metrics": null,
"contains_heads": [
"plain",
"depth"
],
"base_checkpoint_sha256": "ef52456da6bb47e4f38c7cfb26e5eaf88c02895bf674487febd04d24d0dc2059",
"base_model_tensor_sha256": "8c1549a03330f4448146bdb0986e7fa7895c8ccaf15d1e903fe6106a487dc314",
"base_asset_id": "rgbd_e15"
}
],
"code_repository": "YuryLee/DPSeg",
"host": "huggingface",
"repo_id": "Lemma456/DPSeg",
"revision": "weights-v1",
"repository_url": "https://huggingface.co/Lemma456/DPSeg",
"private": false
}