| { | |
| "backbone": "OpenGVLab/pvt_v2_b3", | |
| "count_cls_weight": 0.1, | |
| "density_l1_weight": 0.01, | |
| "density_sigma": 3.0, | |
| "enforce_count": true, | |
| "label_dict": { | |
| "Tree": 0 | |
| }, | |
| "losses": [ | |
| "count", | |
| "count_cls", | |
| "ot", | |
| "density_l1" | |
| ], | |
| "mae_weight": 1.0, | |
| "norm_cood": true, | |
| "num_classes": 1, | |
| "num_of_iter_in_ot": 100, | |
| "ot_weight": 0.1, | |
| "score_integration_radius": 2, | |
| "score_thresh": 0.15, | |
| "sinkhorn_reg": 1.0 | |
| } | |