!33 Optimize the execution time of test case test_rgb_hsv.py

Merge pull request !33 from qianlong21st/ql/fix_test_rgb_hsv
2020-03-31 09:53:01 +08:00 · 2020-03-31 09:53:01 +08:00 · e9be3483fe
parent c1c8fef9ca 8c88b39da1
commit e9be3483fe
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()