darknet53

holocron.models.darknet53(pretrained=False, progress=True, **kwargs)

Darknet-53 from “YOLOv3: An Incremental Improvement”

Parameters:

• pretrained (bool) – If True, returns a model pre-trained on ImageNet

• progress (bool) – If True, displays a progress bar of the download to stderr

Returns:

classification model

Return type:

torch.nn.Module