OPTIMIZED SINGLE SHOT DETECTOR (SSD) MODEL FOR OBJECT DETECTION

    公开(公告)号:US20240221361A1

    公开(公告)日:2024-07-04

    申请号:US18089663

    申请日:2022-12-28

    Inventor: KeYong YU

    CPC classification number: G06V10/765 G06V10/7715 G06V10/82 G06V2201/07

    Abstract: Systems and methods for optimizing single shot detector (SSD) for object detection are disclosed herein. A system receives an image of a plurality of objects. Further, the system determines a plurality of feature layers and a plurality of feature cell sizes corresponding to the received image, based on an aspect ratio of the received image. Furthermore, the system determines aspect ratio of anchor boxes from trained model file, based on aspect ratio of anchor boxes, position and number of anchor boxes to be tiled in each feature cell of the plurality of feature layers. The size of the one or more anchor boxes corresponds to an anchor box aspect ratio. Additionally, the system assigns the one or more anchor boxes as a horizontal tile or a vertical tile in each feature cell, when the anchor box aspect ratio is less than a first pre-defined threshold value and greater than a second pre-defined threshold value, respectively. Further, the system generates one or more feature maps using an object detection model and a neural network (NN) model. The one or more feature maps comprises one or more feature map tensors. Furthermore, the system generates, for each layer of the one or more feature maps, a prediction tensor of predefined dimension from the one or more feature map tensors, using a prediction convolution layer. Additionally, the system detects and classifies the plurality of objects, based on the generated prediction tensor.

Patent Agency Ranking