AmberzzZZ/backbones

ResNet_RS pre-trained

api5454 opened this issue · 2 comments

Hello, and thank you for this excellent work,
But I was wondering about the possibility of obtaining weights
resnet50_weights_tf_dim_ordering_tf_kernels.h5

If there is no way to get it, can you give some advice about training it on data ImageNet,
For your information, I have downloaded resnet50 standard weights from TF-V1 and TF-v2 and the number of layers mismatch appears,
Or this error<..
ValueError: You are trying to load a weight file containing 107 layers into a model with 139 layers.

and,
Can I take advantage of the following checkpoints?? ,

ImageNet Checkpoints from TF.tpu.official

We release configs and checkpoints of the ResNet-RS model family trained on ImageNet in Tensorflow 1.

Model Input Size V100 Lat (s) TPU Lat (ms) Top-1 Accuracy Download
ResNet-RS-50 160x160 0.31 70 78.8 config | ckpt
ResNet-RS-101 160x160 0.48 120 80.3 config | ckpt
ResNet-RS-101 192x192 0.70 170 81.2 config | ckpt
ResNet-RS-152 192x192 0.99 240 82.0 config | ckpt
ResNet-RS-152 224x224 1.48 320 82.2 config | ckpt
ResNet-RS-152 256x256 1.76 410 83.0 config | ckpt
ResNet-RS-200 256x256 2.86 570 83.4 config | ckpt
ResNet-RS-270 256x256 3.76 780 83.8 config | ckpt
ResNet-RS-350 256x256 4.72 1100 84.0 config | ckpt
ResNet-RS-350 320x320 8.48 1630 84.2 config | ckpt
ResNet-RS-420 320x320 10.16 2090 84.4 config | ckpt

Thank you and good luck to you anyway

as you can see there's a little modification on the network architecture from resnet to resnet_rs(the stem & the se_block).
so it's not possible to load the h5 file directly to this model.
but you can try to load the weights layer by layer(as the work i did in covert* scripts).
currently there's some tf version problem so i'm not able to parse the official ckpts correctly but i'm still working on.
after the work is done the convert script would be pushed :)

Thanks for your advice
I got you
Until this time we will close this..