《Few-Shot Learning by Integrating Spatial and Frequency Representation》代码

论文:原来提取特征都是在空域上用CNN提取特征,本文模型额外使用了离散余弦变换提取频域信息再放入另一个CNN提取特征,两部分的特征放一起 送入分类器
论文源码:https://github.com/xiangyu8/PT-MAP-sf

数据

数据集准备

已CUB数据集为例,如果原来机器上有这个数据集就可以直接在项目根目录下新建Dataset/CUB文件夹,然后将数据集复制过来。如果没有可以运行filelists/CUB/dowmload_CUB.sh进行下载。此时的数据集为:
在这里插入图片描述

c生成json文件
根据README.md里的要求,运行filelists/CUB/make_json.py生成train.json, val.json and novel.json

filelists/CUB/make_json.py代码中需要将第11行注释之后再运行,可以得到train.json, val.json and novel.json
在这里插入图片描述

运行train_dct.py

修改参数

根据README.md里的要求,运行train_dct.py需要传参数

python train_dct.py --dataset [DATASETNAME] --method [METHODNAME] --model WideResNet28_10 --train_aug --dct_status

train_dct.pymain调用了parse_args函数,记得加上filter_size
在这里插入图片描述

根据提示,我们设置成在这里插入图片描述

日志准备

Line584Line609都是一些日志的准备工作

数据加载

Line612Line628准备好了经过tramsform变幻的base_loader base_loader_test val_loaderval_loader5way1shot的设置
在这里插入图片描述

训练开始

Line 663调用train_base函数开始训练模型

前向传播

Line69开始前向传播
input_dct_var.shape[13,24,56,56]
在这里插入图片描述

加载val_loader

刚才说val_loader5way1shot的,EpisodicBatchSampler会j将n_classes个类打乱,选取前n_way
在这里插入图片描述

  • 2
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 10
    评论
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值