Channel-Pruning-for-Change-Detection

对遥感图像任务——变化检测（CD）进行结构化通道剪枝

Pruning

运行 prune 文件夹中的 jupyter 文件，对 resnet50 进行通道剪枝

python train.py --backbone 'resnet' --dataset 'levir' --mode 'rsp_300'

python eval.py --backbone 'resnet' --dataset 'levir' --mode 'rsp_300' --path [model path]

python visualization.py --backbone 'resnet' --dataset 'levir' --mode 'rsp_300' --path [model path]

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
models		models
prune		prune
utils		utils
README.md		README.md
eval.py		eval.py
train.py		train.py
visualization.py		visualization.py