提交 e4e9450e 编写于 作者: M minqiyang

Fix random crop op problem

上级 76ee482e
...@@ -21,7 +21,7 @@ from op_test import OpTest ...@@ -21,7 +21,7 @@ from op_test import OpTest
class TestRandomCropOp(OpTest): class TestRandomCropOp(OpTest):
def setUp(self): def setUp(self):
to_crop = np.array([[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]] * to_crop = np.array([[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]] *
5).astype("float32") 5).astype(np.int32)
self.possible_res = [ self.possible_res = [
np.array([[1, 2, 3], [5, 6, 7]]), np.array([[2, 3, 4], [6, 7, 8]]), np.array([[1, 2, 3], [5, 6, 7]]), np.array([[2, 3, 4], [6, 7, 8]]),
np.array([[5, 6, 7], [9, 10, 11]]), np.array([[5, 6, 7], [9, 10, 11]]),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册