zhanghang1989 / RFConv

Rectified Convolution
https://hangzhang.org/RectifiedConv/RectifiedConv.pdf
Apache License 2.0
44 stars 5 forks source link

PyPI PyPI Pre-release pytest License

RFConv

Rectified Convolution

Pretrained Model

model baseline rectified
ResNet-50 76.66 77.10
ResNet-101 78.13 78.74
ResNeXt-50_32x4d 78.17 78.48
ResNeSt-50_2s8x 78.73 79.38

Verify Models:

Prepare ImageNet dataset:

cd scripts/dataset/
# assuming you have downloaded the dataset in the current folder
python prepare_imagenet.py --download-dir ./

Test Model

# use resnest50 as an example
cd scripts/
python verify.py --model resnet50 --crop-size 224