108 Star 869 Fork 1.5K

MindSpore/models

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

ECENet

paper link: http://arxiv.org/abs/2308.12894

Requirements

conda create -n ms2.0.0 python=3.7
conda activate ms2.0.0
export MS_VERSION=2.0.0
pip install -r requirements.txt

Inference

python main.py ./ckpt/sdt_swin-base-4-12_640x640_ade20k_1236length_ms_total-ms.py ./ckpt/ms_convert.ckpt ./ADEChallengeData2016 --eval mIoU mFscore mDice

  • Results
aAcc mIoU mAcc mFscore mPrecision mRecall mDice
84.57 54.17 65.46 67.68 72.03 65.46 67.68

Citation

@inproceedings{liu2023boosting,
  title={Boosting Semantic Segmentation from the Perspective of Explicit Class Embeddings},
  author={Liu, Yuhe and Liu, Chuanjian and Han, Kai and Tang, Quan and Qin, Zengchang},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={821--831},
  year={2023}
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/models.git
git@gitee.com:mindspore/models.git
mindspore
models
models
master

搜索帮助