项目简介
本项目运用Deep - DAP模型开展零样本学习相关研究,借助CNN模型提取图像特征,通过数据集训练多个属性分类器,以解决零样本学习问题,最终实现对动物属性的有效识别。
项目的主要特性和功能
- 数据处理:提供脚本读取AwA和CUB数据集相关信息,将图像数据与标签保存为便于使用的格式。
- 模型定义:定义了不同CNN下的Deep - DAP模型,用于后续训练。
- 模型训练:通过训练脚本,训练Deep - DAP模型并保存训练得到的模型。
- 模型可视化:可对模型的权重进行可视化,辅助理解模型工作原理。
安装使用步骤
- 已下载本项目源码文件后,确保Python版本为3.6。
- 安装所需框架和库,包括tensorflow、keras、kera - vis。
- 根据实际数据集路径修改代码中对应路径。
- 运行datatool.py读取数据集相关信息。
- 运行Deep - DAP.py训练模型。
- (可选)运行model - vis.py对模型的权重进行可视化。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】