Streaming Classification support (#9106)
* Streaming Classification support * Streaming Classification support * Streaming Classification support
This commit is contained in:
1 parent
48e56d3c9b
commit
51c9f92297
3 files changed
+131
-40
No files matched your search
@@ -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)])
|
||||
Reference in new issue
Block a user