cuibinge commited on
Commit
aec8f5e
·
verified ·
1 Parent(s): dc8de42

Align code with polygon-only training

Browse files
Files changed (1) hide show
  1. configs/sampoly_polygon_head.json +45 -0
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
+ }