Computer Vision models zoo¶

In [ ]:

from fastai.gen_doc.nbdoc import *
from fastai.vision.models.darknet import Darknet
from fastai.vision.models.wrn import wrn_22, WideResNet

On top of the models offered by torchivision, the fastai library has implementations for the following models:

Darknet architecture, which is the base of Yolo v3
Unet architecture based on a pretrained model. The original unet is described here, the model implementation is detailed in models.unet
Wide resnets architectures, as introduced in this article.

In [ ]:

show_doc(Darknet, doc_string=False)

`class` `Darknet`[source]

Darknet(num_blocks:Collection[int], num_classes:int, nf=32) :: Module

Create a Darknet with blocks of sizes given in num_blocks, ending with num_classes and using nf initial features. Darknet53 uses num_blocks = [1,2,8,8,4].

In [ ]:

show_doc(WideResNet, doc_string=False)

`class` `WideResNet`[source]

WideResNet(num_groups:int, N:int, num_classes:int, k:int=1, drop_p:float=0.0, start_nf:int=16) :: Module

Create a wide resnet with blocks num_groups groups, each containing blocks of size N. k is the width of the resnet, start_nf the initial number of features. Dropout of drop_p is applied at the end of each block.

In [ ]:

show_doc(wrn_22)

`wrn_22`[source]

wrn_22()

Creates a wide resnet for CIFAR-10 with num_groups=3, N=3, k=6 and drop_p=0..

Computer Vision models zoo¶

class Darknet[source]

class WideResNet[source]

wrn_22[source]

`class` `Darknet`[source]

`class` `WideResNet`[source]

`wrn_22`[source]