ssds.modeling

• ssds.modeling.layers
• ssds.modeling.nets
• ssds.modeling.ssds

ssds.modeling.model_builder

ssds.modeling.model_builder.create_anchors(cfg, model, image_size, visualize=False)

current version for generate the anchor, only generate the default anchor for each feature map layers :returns: OrderedDict(key=stride, value=default_anchors) :rtype: anchors

ssds.modeling.model_builder.create_decoder(cfg)

Generate decoder based on the cfg.POST_PROCESS.

The generated decoder is the object of class Decoder, check more details by ssds.modeling.layers.decoder.Decoder.

Parameters

cfg – defined cfg.POST_PROCESS

ssds.modeling.model_builder.create_model(cfg)

create the model based on the config files :returns: torch ssds model with backbone as net