interpolations added (#693)

bf3ab297 · surgan12 · Soumith Chintala · 59802fa1 · bf3ab297
Commit bf3ab297 authored Dec 21, 2018 by surgan12 Committed by Soumith Chintala Dec 20, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

torchvision/transforms/transforms.py torchvision/transforms/transforms.py +2 -0

No files found.
--- a/torchvision/transforms/transforms.py
+++ b/torchvision/transforms/transforms.py
@@ -34,6 +34,8 @@ _pil_interpolation_to_str = {
    Image.BILINEAR: 'PIL.Image.BILINEAR',
    Image.BICUBIC: 'PIL.Image.BICUBIC',
    Image.LANCZOS: 'PIL.Image.LANCZOS',
+    Image.HAMMING: 'PIL.Image.HAMMING',
+    Image.BOX: 'PIL.Image.BOX',
 }