What are models trained on?

[mazatov]

What are the models trained on? I see we can put imagenet as a parameter for weights. Do they have 3D stacks in imagenet?

[ZFTurbo]

Initial models were trained on imagenet. Then imagenet weights were converted from 2D to 3D format.

[mazatov]

What algorithm are you using to inflate the weights from 2D to 3D?

[ZFTurbo]

Description of method can be found here:
https://arxiv.org/abs/2104.01687

Code for conversion here:
https://github.com/ZFTurbo/classification_models_3D/blob/master/convert_imagenet_weights_to_3D_models.py