diff --git a/test.py b/test.py index 3f15271..3ed7d2a 100644 --- a/test.py +++ b/test.py @@ -96,7 +96,7 @@ def test(args): image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) height, width = image.shape[:2] - points = [[0,0], [0,width-1], [height-1,width-1], [height-1,0]] + points = [[0,0], [width-1,0], [width-1,height-1], [0,height-1]] image = preprocess(image, points, cfg.image_size) image = np.expand_dims(image, 0) @@ -119,4 +119,4 @@ def test(args): args = parser.parse_args() test(args) - \ No newline at end of file +