# SegWorker **Repository Path**: xizeer/seg-worker ## Basic Information - **Project Name**: SegWorker - **Description**: 自行搭建的语义分割框架 - **Primary Language**: Python - **License**: MulanPSL-2.0 - **Default Branch**: master - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 3 - **Forks**: 0 - **Created**: 2022-08-12 - **Last Updated**: 2024-12-06 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README # SegWorker #### 介绍 自行搭建的语义分割训练框架,主要参照mmseg的架构,对一些配置进行了一些优化。目前还在测试中... #### 软件架构 Software architecture description 1. config 配置文件 2. data 数据集加载模块 3. model 模型机构结构 4. tools 乱七八糟工具包如:labelme,精灵标注助手的标签转化 5. trainer 模型训练对象 6. utils 模型导出工具包 #### 安装环境 pip install -r requirements.txt #### 使用实例 1. 训练模型 `python train.py -c config/default.yml` 2. 模型导出 `python utils/exports.py -c config_path -m model_path -o output_path` #### 未来改进方向 1.新增coco等常用语义分割数据集的加载模块 2.引入更多的模型并各个模型的baseline 3.加入模型推理速度等测试工具模块