def setUp(self):
chainer.set_debug(True)
np.random.seed(0)
x = np.random.randint(0, 255, size=(224, 224, 3)).astype(np.float)
x -= np.array([[[102.9801, 115.9465, 122.7717]]])
self.x = np.expand_dims(x, 0).transpose(0, 3, 1, 2).astype(np.float32)
self.im_info = np.array([[224, 224, 1.6]])
self.gt_boxes = np.array([
[10, 10, 60, 200, 0],
[50, 100, 210, 210, 1],
[160, 40, 200, 70, 2]
])
评论列表
文章目录