Streaming Classification support (#9106)

* Streaming Classification support

* Streaming Classification support

* Streaming Classification support
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-08-23 17:54:51 +02:00
1 parent 48e56d3c9b
commit 51c9f92297
3 files changed
+131 -40

No files matched your search

+1
View File
@@ -344,4 +344,5 @@ def classify_albumentations(augment=True,
def classify_transforms(size=224):
# Transforms to apply if albumentations not installed
assert isinstance(size, int), f'ERROR: classify_transforms size {size} must be integer, not (list, tuple)'
return T.Compose([T.ToTensor(), T.Resize(size), T.CenterCrop(size), T.Normalize(IMAGENET_MEAN, IMAGENET_STD)])