def clip_boxes(boxes, im_shape, gpu=-1):
if gpu >= 0:
with cuda.Device(gpu):
return _clip_boxes(boxes, im_shape)
else:
return _clip_boxes(boxes, im_shape)
微信
公众号
扫码关注公众号