EfficientNet-PyTorch-3D icon indicating copy to clipboard operation
EfficientNet-PyTorch-3D copied to clipboard

.from_pretrained for 3d inputs

Open mahootiha-maryam opened this issue 2 years ago • 1 comments

Cant we use pretrained models for 3d images? I had 3d clinical images they are not too much and I wanted to use them for image classification. so I needed a kind of pretrained model. I saw repository but I couldnt understand can I use .from_pretrained for 3d or not. If it is possible can you give example by code?

mahootiha-maryam avatar Aug 01 '22 09:08 mahootiha-maryam

class EfficientNet3D(nn.Module):
    """
    An EfficientNet model. Most easily loaded with the .from_name or .from_pretrained methods
    Args:
        blocks_args (list): A list of BlockArgs to construct blocks
        global_params (namedtuple): A set of GlobalParams shared between blocks
    Example:
        model = EfficientNet3D.from_pretrained('efficientnet-b0')

You wrote in class EfficientNet3D but when I use this code model = EfficientNet3D.from_pretrained('efficientnet-b0') I face this error type object 'EfficientNet3D' has no attribute 'from_pretrained'

mahootiha-maryam avatar Aug 01 '22 09:08 mahootiha-maryam