diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-21 20:00:51 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-21 20:00:51 +0800 |
| commit | 315495ffe14986cfce90e92203cc317eef5ba5cf (patch) | |
| tree | d261a573117d405a5b3faabd0dc549e95a1cde7b /requirements.txt | |
| parent | 0d4c20a8104a77f2f61f5ce0ac46c3a26d61b9c1 (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 'requirements.txt')
| -rw-r--r-- | requirements.txt | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/requirements.txt b/requirements.txt index a0eaf1a..c58a9c3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,6 @@ torch~=1.7.1 torchvision~=0.8.0a0+ecf4e9c numpy~=1.19.4 -tqdm~=4.54.1
\ No newline at end of file +tqdm~=4.54.1 +Pillow~=8.0.1 +matplotlib~=3.3.3
\ No newline at end of file |