Unverified Commit c9eab681 authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Add torchscript test for decode_gif (#8418)

parent 17d972ac
......@@ -552,7 +552,8 @@ def test_pathlib_support(tmpdir):
@pytest.mark.parametrize("name", ("gifgrid", "fire", "porsche", "treescap", "treescap-interlaced", "solid2", "x-trans"))
def test_decode_gif(tmpdir, name):
@pytest.mark.parametrize("scripted", (True, False))
def test_decode_gif(tmpdir, name, scripted):
# Using test images from GIFLIB
# https://sourceforge.net/p/giflib/code/ci/master/tree/pic/, we assert PIL
# and torchvision decoded outputs are equal.
......@@ -565,7 +566,9 @@ def test_decode_gif(tmpdir, name):
with open(path, "wb") as f:
f.write(requests.get(url).content)
tv_out = read_image(path)
encoded_bytes = read_file(path)
f = torch.jit.script(decode_gif) if scripted else decode_gif
tv_out = f(encoded_bytes)
if tv_out.ndim == 3:
tv_out = tv_out[None]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment