From 8a6c1d4285adf310784adbe1be94c1709215cbd7 Mon Sep 17 00:00:00 2001 From: zhang Date: Sat, 4 Jun 2022 08:53:06 +0800 Subject: 0604 --- pretrained/swin/demo.py | 1 + 1 file changed, 1 insertion(+) 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( [ -- cgit v1.2.3