00c06ff2b04c3c991b6fbe1281ff3eede9ca4f94,gluon/gluoncv2/models/darknet.py,DarkNet,__init__,#DarkNet#Any#Any#Any#Any#Any#Any#Any#,156

Before Change

                kernel_size=1,
                in_channels=in_channels))
            if cls_activ:
                self.output.add(nn.LeakyReLU(alpha=0.1))
            self.output.add(nn.AvgPool2D(
                pool_size=avg_pool_size,
                strides=1))

After Change

def darknet_tiny(**kwargs):
    
    DarkNet Tiny model from "Darknet: Open source neural networks in c," https://github.com/pjreddie/darknet.

    Parameters:
    ----------
    pretrained : bool, default False
        Whether to load the pretrained weights for model.
    ctx : Context, default CPU
        The context in which to load the pretrained weights.
    root : str, default "~/.mxnet/models"
        Location for keeping the model parameters.
    
    return get_darknet(version="tiny", model_name="darknet_tiny", **kwargs)

Instances