图像分类辅助标注：决策树辅助

阅读量：

目录

引言
- 构建决策树模型的训练过程
- 对决策树模型进行推理操作

前言

在进行图像分类标注任务时，首先对部分数据进行标注操作，随后提取这些图像的特征信息，如【代码篇】图像预处理阶段：提取图像特征所示。通过结合所提取的特征信息与对应的分类标签，构建一个决策树模型。之后，借助该模型对剩余数据进行分类处理，从而有效降低人工标注的工作量。

训练决策树模型

复制代码

    import os
    import pandas as pd
    from sklearn import tree
    import graphviz
    import matplotlib.pyplot as plt
    from sklearn.tree import DecisionTreeClassifier
    from sklearn.externals import joblib
    
    import utils
    
    
    root_dir = "../Image-Downloader-master/download_images/gan/emoji_combine"
    excel_path = "img_combine_info_type.xls"
    data = pd.read

全部评论 (0)

还没有任何评论哟~