# shayan **Repository Path**: fumenghao/shayan ## Basic Information - **Project Name**: shayan - **Description**: No description available - **Primary Language**: Unknown - **License**: AFL-3.0 - **Default Branch**: master - **Homepage**: None - **GVP Project**: No ## Statistics - **Stars**: 0 - **Forks**: 0 - **Created**: 2024-12-15 - **Last Updated**: 2024-12-15 ## Categories & Tags **Categories**: Uncategorized **Tags**: None ## README # 砂岩图像分割项目 ## 项目简介 本项目旨在通过机器学习方法对砂岩图像进行自动分割。项目使用了随机森林分类器来识别砂岩图像中的不同区域。 ## 项目效果 ![输入图片说明](segment_image.png) ## 功能特性 - **图像预处理**:将图像转换为灰度图,应用Sobel、高斯和均值滤波器。 - **特征提取**:从图像中提取RGB通道,以及经过不同滤波器处理后的特征。 - **模型训练**:使用随机森林分类器对图像进行训练和分类。 - **性能评估**:计算模型的准确率,并展示分割结果。 ## 技术栈 - **Python**:编程语言 - **NumPy**:数值计算库 - **Matplotlib**:绘图库 - **scikit-image**:图像处理库 - **scikit-learn**:机器学习库 - **joblib**:模型序列化库 - **tqdm**:进度条库 ## 安装与部署 ### 安装步骤 1. 克隆代码库: ```bash 安装依赖: bash pip install numpy matplotlib scikit-image scikit-learn joblib tqdm 准备图像文件: Sandstone_1.tif Sandstone_1_segment.tif Sandstone_2.tif Sandstone_2_segment.tif 部署步骤 运行create_clf.py来训练模型并保存: python create_clf.py 运行segment.py来进行图像分割: python segment.py 使用说明 运行segment.py脚本将自动加载训练好的模型,对Sandstone_2.tif图像进行分割,并显示分割结果。