EfficientNet-PyTorch-3D
EfficientNet-PyTorch-3D copied to clipboard
.from_pretrained for 3d inputs
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?
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'