fasterRCNN.py 文件源码

def classify(self,x=None):
        if x is None:
            x=Tensor.context

        if not isinstance(x,ImageTensor):
            x=Input(x)

        xp = Deel.xp
        x_data = xp.asarray(self.x_batch)
        xv = chainer.Variable(x.value, volatile=True)

        h, w = xv.data.shape[2:]
        cls_score, bbox_pred  = self.func(xv,np.array([[h, w, x.im_scale]]))
        draw_rois(x.content,x.im_scale,self.func.rois,bbox_pred,cls_score.data)

        if Deel.gpu >= 0:
            cls_score = chainer.cuda.cupy.asnumpy(cls_score)
            bbox_pred = chainer.cuda.cupy.asnumpy(bbox_pred)
        result = draw_result(x.content, 1.0, cls_score.data, bbox_pred,0.3,0.8)
        cv.imshow("res",result)
        cv.waitKey(0)