CloserLookFewShot 项目使用教程

CloserLookFewShot 项目使用教程

CloserLookFewShot source code to ICLR'19, 'A Closer Look at Few-shot Classification' 项目地址: https://gitcode.com/gh_mirrors/cl/CloserLookFewShot

1. 项目目录结构及介绍

CloserLookFewShot/
├── data/
│   ├── filelists/
│   │   ├── CUB/
│   │   ├── miniImagenet/
│   │   ├── omniglot/
│   │   └── emnist/
│   ├── methods/
│   ├── record/
│   └── save_features/
├── backbone.py
├── configs.py
├── io_utils.py
├── LICENSE.txt
├── README.md
├── save_features.py
├── test.json
├── test.py
├── train.py
└── utils.py

目录结构说明

• data/: 存放数据集相关文件，包括数据集的文件列表、方法实现、记录文件和保存的特征文件。

  • filelists/: 包含不同数据集的文件列表，如CUB、miniImagenet、omniglot和emnist。
  • methods/: 存放各种方法的实现代码。
  • record/: 存放实验结果记录文件。
  • save_features/: 存放保存的特征文件。

• backbone.py: 定义了模型的骨干网络。

• configs.py: 项目的配置文件，包含各种参数设置。

• io_utils.py: 输入输出工具，用于处理数据输入和输出。

• LICENSE.txt: 项目的许可证文件。

• README.md: 项目的说明文档。

• save_features.py: 用于保存提取的特征文件。

• test.json: 测试配置文件。

• test.py: 测试脚本，用于运行测试。

• train.py: 训练脚本，用于训练模型。

• utils.py: 工具函数，包含各种辅助功能。

2. 项目启动文件介绍

train.py

train.py 是项目的启动文件，用于训练模型。可以通过命令行参数指定数据集、模型和方法等。

使用示例：

python train.py --dataset miniImagenet --model Conv4 --method baseline --train_aug

test.py

test.py 用于测试训练好的模型。可以通过命令行参数指定数据集、模型和方法等。

使用示例：

python test.py --dataset miniImagenet --model Conv4 --method baseline --train_aug

save_features.py

save_features.py 用于保存提取的特征文件，以提高测试速度。

使用示例：

python save_features.py --dataset miniImagenet --model Conv4 --method baseline --train_aug

3. 项目的配置文件介绍

configs.py

configs.py 是项目的配置文件，包含了各种参数设置，如数据集路径、模型参数、训练参数等。

主要配置项：

• data_dir: 数据集路径。
• model_dir: 模型保存路径。
• method: 使用的训练方法。
• train_aug: 是否使用数据增强。

示例：

data_dir = {
    'miniImagenet': '/path/to/miniImagenet',
    'CUB': '/path/to/CUB',
    'omniglot': '/path/to/omniglot',
    'emnist': '/path/to/emnist'
}

model_dir = '/path/to/save/models'
method = 'baseline'
train_aug = True

通过修改 configs.py 文件中的配置项，可以自定义项目的运行参数。

CloserLookFewShot source code to ICLR'19, 'A Closer Look at Few-shot Classification' 项目地址: https://gitcode.com/gh_mirrors/cl/CloserLookFewShot