Skip to content

Commit

Permalink
test(gf1): 云适配器在GF1数据集上的测试结果
Browse files Browse the repository at this point in the history
- 在GF1数据集上进行了云适配器模型的测试
- 测试指标:
  - mIoU: 92.55%
  - mDice: 96.02%
  - mFscore: 96.02%
  - mPrecision: 96.05%
  - mRecall: 95.99%
- 测试结果表明模型在GF1数据集上具有较高的准确性
  • Loading branch information
caixiaoshun committed Nov 22, 2024
1 parent d0fb6da commit edea9ff
Show file tree
Hide file tree
Showing 22 changed files with 5,301 additions and 5 deletions.
4 changes: 2 additions & 2 deletions cloud_adapter/datasets/hrc_whu.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,8 @@ class HRCWHUDataset(BaseSegDataset):
palette=[[0, 0, 0],[255, 255, 255]])

def __init__(self,
img_suffix='.tif',
seg_map_suffix='.tif',
img_suffix='.png',
seg_map_suffix='.png',
reduce_zero_label=False,
**kwargs) -> None:
super().__init__(
Expand Down
4 changes: 2 additions & 2 deletions cloud_adapter/models/backbones/vitadapter_dinov2.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,12 @@
from mmseg.models.builder import BACKBONES
# from ops.modules import MSDeformAttn

from rein.models.backbones import DinoVisionTransformer
from cloud_adapter.models.backbones import DinoVisionTransformer
from mmcv.ops import MultiScaleDeformableAttention as MSDeformAttn
from timm.models.layers import DropPath, trunc_normal_
from torch.nn.init import normal_
from .utils import set_requires_grad, set_train
from rein.models.backbones.adapter_modules import SpatialPriorModule, InteractionBlock, deform_inputs
from cloud_adapter.models.backbones.adapter_modules import SpatialPriorModule, InteractionBlock, deform_inputs

_logger = logging.getLogger(__name__)

Expand Down
2 changes: 1 addition & 1 deletion cloud_adapter/utils/init_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from mmengine.runner import load_checkpoint
from mmseg.registry import MODELS
from mmseg.utils import SampleList, dataset_aliases, get_classes, get_palette
from rein.hooks.load_backbone_hook import load_backbone
from cloud_adapter.hooks.load_backbone_hook import load_backbone
import torch

def init_model(config: Union[str, Path, Config],
Expand Down
Loading

0 comments on commit edea9ff

Please sign in to comment.