summaryrefslogtreecommitdiff
path: root/models/layers.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2020-12-21 20:00:51 +0800
committerJordan Gong <jordan.gong@protonmail.com>2020-12-21 20:00:51 +0800
commit315495ffe14986cfce90e92203cc317eef5ba5cf (patch)
treed261a573117d405a5b3faabd0dc549e95a1cde7b /models/layers.py
parent0d4c20a8104a77f2f61f5ce0ac46c3a26d61b9c1 (diff)
Change image loading technique
1. Use Pillow.Image.open instead of torchvision.io.read_image to read image 2. Transforming PIL images instead of tensors which performs better and device option is removed 3. Images are normalized now
Diffstat (limited to 'models/layers.py')
0 files changed, 0 insertions, 0 deletions