diff options
| author | zhang <zch921005@126.com> | 2022-06-04 08:53:06 +0800 |
|---|---|---|
| committer | zhang <zch921005@126.com> | 2022-06-04 08:53:06 +0800 |
| commit | 8a6c1d4285adf310784adbe1be94c1709215cbd7 (patch) | |
| tree | 2e62d9f28189141be2c0ce1852e40c02121b1444 | |
| parent | 44097da7288042e988bcb89f1c6cc817a8e1eec9 (diff) | |
0604
| -rw-r--r-- | pretrained/swin/demo.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/pretrained/swin/demo.py b/pretrained/swin/demo.py index 6f8d87b..6e12475 100644 --- a/pretrained/swin/demo.py +++ b/pretrained/swin/demo.py @@ -5,6 +5,7 @@ from PIL import Image from torchvision import transforms as T import json + if __name__ == '__main__': trans_ = T.Compose( [ |