p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Optimize the execution time of test case test_rgb_hsv.py

This commit is contained in:
qianlong 2020-03-30 20:16:45 +08:00
parent f1b722297e
commit 8c88b39da1
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/ut/python/dataset/test_rgb_hsv.py
View File

@ -133,6 +133,7 @@ def test_rgb_hsv_pipeline():
# First dataset # First dataset
transforms1 = [ transforms1 = [
vision.Decode(), vision.Decode(),
vision.Resize([64, 64]),
vision.ToTensor() vision.ToTensor()
] ]
transforms1 = vision.ComposeOp(transforms1) transforms1 = vision.ComposeOp(transforms1)
@ -142,6 +143,7 @@ def test_rgb_hsv_pipeline():
# Second dataset # Second dataset
transforms2 = [ transforms2 = [
vision.Decode(), vision.Decode(),
vision.Resize([64, 64]),
vision.ToTensor(), vision.ToTensor(),
vision.RgbToHsv(), vision.RgbToHsv(),
vision.HsvToRgb() vision.HsvToRgb()