|
|
@ -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()
|
|
|
|