Align code with polygon-only training
Browse files
configs/sampoly_polygon_head.json
ADDED
|
@@ -0,0 +1,45 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "marine_sampoly_polygon",
|
| 3 |
+
"purpose": "Typical marine ecological feature polygon extraction",
|
| 4 |
+
"encoder": {
|
| 5 |
+
"type": "DINOv3 ViT + DINOv3 ConvNeXt dual branch",
|
| 6 |
+
"fusion": "Coast-style cross-scale semantic/detail fusion",
|
| 7 |
+
"fine_tuning": {
|
| 8 |
+
"vit_unfrozen_blocks": 2,
|
| 9 |
+
"convnext_unfrozen_stages": 2,
|
| 10 |
+
"unfreeze_all_norm_layers": true
|
| 11 |
+
}
|
| 12 |
+
},
|
| 13 |
+
"head": {
|
| 14 |
+
"style": "SAMPolyBuild-like polygon mapping",
|
| 15 |
+
"not_a_box_detector": true,
|
| 16 |
+
"outputs": [
|
| 17 |
+
"foreground_mask_logits",
|
| 18 |
+
"boundary_logits",
|
| 19 |
+
"vertex_logits",
|
| 20 |
+
"polygon_objectness",
|
| 21 |
+
"fixed_vertex_polygons"
|
| 22 |
+
],
|
| 23 |
+
"default_num_queries": 100,
|
| 24 |
+
"default_vertices_per_polygon": 32,
|
| 25 |
+
"supervision": [
|
| 26 |
+
"mask",
|
| 27 |
+
"boundary",
|
| 28 |
+
"vertex_heatmap",
|
| 29 |
+
"polygon_sequence"
|
| 30 |
+
]
|
| 31 |
+
},
|
| 32 |
+
"data": {
|
| 33 |
+
"preferred_formats": [
|
| 34 |
+
"images/{split} + masks/{split}",
|
| 35 |
+
"images/{split} + annotations/{split}.json",
|
| 36 |
+
"images/{split} + annotations/ann.json"
|
| 37 |
+
],
|
| 38 |
+
"polygon_annotation_format": "COCO segmentation polygons",
|
| 39 |
+
"bbox_only_data": "not sufficient for true polygon training; reject from the main dataset"
|
| 40 |
+
},
|
| 41 |
+
"notes": [
|
| 42 |
+
"Training this head requires masks or polygon annotations; rectangle boxes cannot teach true boundaries.",
|
| 43 |
+
"Do not assume external coastline, land mask, or GIS vector layers are available."
|
| 44 |
+
]
|
| 45 |
+
}
|