Res2Net / Res2Net-PretrainedModels

(ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"
https://mmcheng.net/res2net/
1.07k stars 215 forks source link

Can you share codes of training and data augmentation? #13

Closed c747903646 closed 4 years ago

c747903646 commented 4 years ago

Hi,

I hope to reproduce the same training result by runing program on my machine. Can you share codes of training and data augmentation that were used to get the result reported in this paper?

gasvn commented 4 years ago

Please refer to the offical pytorch training code: https://github.com/pytorch/examples/tree/master/imagenet

c747903646 commented 4 years ago

These 3 py files didn't include all model reported in paper, such as Res2NeXt-29, 8c×25w×4s. Can I implement this model by adding codes as below to the end of res2next.py? Is there anything else I should do?

def res2next29_8cx25wx4scale(pretrained=False, **kwargs): model = CifarRes2NeXt(Bottle2neckX, layers = [3, 3, 3], baseWidth = 25, cardinality=8, scale = 4, num_classes=100) return model

gasvn commented 4 years ago

The cifar-res2next is based on cifar-resnext. The implemention of cifar-resnext is different from ImageNet-resnext, as described in the resnext paper. And you can replace the bottleneck block in cifar-resnext with the res2next block.

c747903646 commented 4 years ago

OK, thanks!