IC Conv:pytorch正式实施“具有有效膨胀搜索的初始卷积”(CVPR 2021口头) 源码
转换 该存储库是的正式实现。 入门 下载ImageNet预训练的。 解压缩文件以获取以下目录树 |-- README.md |-- ckpt | |-- detection | |-- human_pose | |-- segmentation |-- config |-- model |-- pattern_zoo 使用方便 当前的实现与特定的下游任务相关。 用户可以通过以下简单方法快速使用IC-Conv。 from models import IC_ResNet import torch net = IC_ResNet ( depth = 50 , pattern_path = 'pattern_zoo/detection/ic_r50_k9.json' ) net . eval () inputs = torch . rand ( 1 , 3 , 32 , 32 )
用户评论