CPC G06N 3/088 (2013.01) [G06N 3/045 (2023.01); G06N 3/048 (2023.01)] | 16 Claims |
1. A method for object detection, the method comprises:
calculating, by a teacher object detection neural network (ODNN), weights to be applied during a calculation of a weighted sum;
training a student object detection neural network (ODNN) to mimic a teacher ODNN;
wherein the training comprises calculating a teacher student detection loss that is based on a pre-bounding-box output of the teacher ODNN that is a weighted sum of pre-bounding-box outputs of different ODNNs that belong to the teacher ODNN; and
detecting one or more objects in an image; wherein the detecting comprises:
feeding the image to the trained student ODNN;
outputting by the trained student ODNN a student pre-bounding-box output; and
calculating one or more bounding boxes based on the student pre-bounding-box output.
|