Optimize the execution time of test case test_rgb_hsv.py

5 years ago · 8c88b39da1
parent f1b722297e
commit 8c88b39da1
1 changed files with 2 additions and 0 deletions
--- a/tests/ut/python/dataset/test_rgb_hsv.py
+++ b/tests/ut/python/dataset/test_rgb_hsv.py
@ -133,6 +133,7 @@ def test_rgb_hsv_pipeline():
    # First dataset
    transforms1 = [
        vision.Decode(),
+        vision.Resize([64, 64]),
        vision.ToTensor()
    ]
    transforms1 = vision.ComposeOp(transforms1)
@ -142,6 +143,7 @@ def test_rgb_hsv_pipeline():
    # Second dataset
    transforms2 = [
        vision.Decode(),
+        vision.Resize([64, 64]),
        vision.ToTensor(),
        vision.RgbToHsv(),
        vision.HsvToRgb()