AlexNet
Our implementation of ConvNet is based on DC.
We provide the following interface to initialize a AlexNet model:
dd_ranking.utils.get_alexnet(model_name: str, im_size: tuple, channel: int, num_classes: int, pretrained: bool, model_path: str) [SOURCE]
Parameters
- model_name(str): Name of the model. Please navigate to models for the model naming convention in DD-Ranking.
- im_size(tuple): Image size.
- channel(int): Number of channels of the input image.
- num_classes(int): Number of classes.
- pretrained(bool): Whether to load pretrained weights.
- model_path(str): Path to the pretrained model weights.