ssds.pytorch API

• ssds.core
• ssds.dataset
• ssds.modeling
• ssds.pipeline

ssds.ssds

class ssds.ssds.SSDDetector(cfg_file, is_print=False)

The class used for ssds model inference

Parameters

• cfg_file (str) – The path to the config file

• is_print (bool) – Whether to print the model arch and the anchor strategy

__call__(imgs)

Applied the SSD model to do the inference for the given images

Parameters

imgs (np.ndarray) – The images used for inference, with shape [H,W,3], [3,H,W] [N,H,W,3] or [N,3,H,W]