FBNetV3

FBNetV3 makes up a family of state-of-art compact neural networks that is generated through Network Architecture Recipe Search, NARS. NARS is an advanced version of Network Architecture Search that searches for both the architecture and the training recipes. FBNetV3 has been shown to improve the mAP (mean Average Precision).

Parameters

Backbone Network

Many image segmentation or object detection task use feature extraction and use of regional proposals as it was proven to be more cost effective. Therefore, FBNetV3 has similar backbone network at the beginning to extract such features.

Pooler Resolution (Mask)

It is the size to pool proposals before feeding them to the mask predictor, in model playground the default value is set as 14.

Pooler Resolution (Box)

It is the size to pool proposals before feeding them to the mask predictor, in model playground the default value is set as 6.

Weights

Before the training process, the weights in the neural network has to be initialized to a certain values. The users will initialize the weights to FBNetV3a-DSMask-C4 COCO.

IOU Threshold

The IOU threshold is used to decide whether the bounding box contains a background or an object.

Everything above the value of the upper bound will be classified as objects and everything lower than the lower bound will be classified as background. The values in between the lower and the upper bound are ignored.

Normalization method

Normalization techniques help to decrease the overall training time of the model. It makes the contribution of the features uniform by normalizing the weights. This also helps to avoid the weights from exploding and hence makes the optimization faster.

There are three available normalization method in model playground:

• GN
• SyncBN
• naiveSyncBN

SyncBN

In this normalization techniques, where the weights are scaled and shifted by the variance and the mean. Mathematically, it is given as,

$$\hat{x}=\frac{x-E(x)}{\sqrt{Var(x)}-\epsilon}\
y=\gamma \cdot \hat{x}+\beta$$

The mean and standard-deviation are calculated per-dimension over all mini-batches of the same process groups. Later again, the scaling and shifting happens with other two constants: €€\gamma€€ and €€\beta€€. These are hyperparameters, and are usually learnable through the network.

NaiveSyncBN

In this normalization technique the weights are assigned equally to all the images regardless of their dimension. With this, we reduce the need to accurately compute mean and variance for each of the batches. A little difference has been observed between such simplified calculation and accurate mean and variance calculation.

GN

Group Batch normalization, abbreviated as GN, is another normalization technique that normalizes a group of parameters. If the input dimension is 50, them the GN normalization can group those 50 parameters in a group of 5, and normalize each group with its own mean and variance.

Pre NMS number of proposals

It is the maximum number of the proposals to be considered before the non maximal suppression. The proposals are sorted descending after confidence and only the ones with the highest confidence are chosen.

Post NMS number of proposals

It is the maximum number of proposals to be considered after the non maximal suppression. The probability of detecting more objects is high if this number is high but the computation cost is also increased since more regional proposals has to be processed.

Code Implementation

import urllib

import torch
from mobile_cv.model_zoo.models.fbnet_v2 import fbnet
from mobile_cv.model_zoo.models.preprocess import get_preprocess
from PIL import Image


def _get_input():
    # Download an example image from the pytorch website
    url, filename = (
        "https://github.com/pytorch/hub/raw/master/dog.jpg",
        "dog.jpg",
    )
    local_filename, headers = urllib.request.urlretrieve(url, filename)
    input_image = Image.open(local_filename)
    return input_image


def run_fbnet_v2():
    # fbnet models, supported models could be found in
    # mobile_cv/model_zoo/models/model_info/fbnet_v2/*.json
    model_name = "dmasking_l3"

    # load model
    model = fbnet(model_name, pretrained=True)
    model.eval()
    preprocess = get_preprocess(model.arch_def.get("input_size", 224))

    # load and process input
    input_image = _get_input()
    input_tensor = preprocess(input_image)
    input_batch = input_tensor.unsqueeze(0)

    # run model
    with torch.no_grad():
        output = model(input_batch)
    output_softmax = torch.nn.functional.softmax(output[0], dim=0)
    print(output_softmax.max(0))


if __name__ == "__main__":
    run_fbnet_v2()