Method predict

examples/python_with_dll/python_trt.py:27–32

(self, img)

                                      num_class)

    def predict(self, img):
        rows, cols = img.shape[0], img.shape[1]
        res_arr = np.zeros((self.max_bbox, 6), dtype=np.float32)
        self.yolo.Detect(self.c_point, c_int(rows), c_int(cols), img.ctypes.data_as(POINTER(c_ubyte)), res_arr)
        self.bbox_array = res_arr[~(res_arr == 0).all(1)]
        return self.bbox_array


class_names = [