forked from mindspore-Ecosystem/mindspore
!33 Optimize the execution time of test case test_rgb_hsv.py
Merge pull request !33 from qianlong21st/ql/fix_test_rgb_hsv
This commit is contained in:
commit
e9be3483fe
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue