Unverified Commit 0692e42f authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

added `panoptic_deeplab_policy`

parent cb7e2650
...@@ -1583,6 +1583,7 @@ class AutoAugment(ImageAugment): ...@@ -1583,6 +1583,7 @@ class AutoAugment(ImageAugment):
'reduced_cifar10': self.policy_reduced_cifar10(), 'reduced_cifar10': self.policy_reduced_cifar10(),
'svhn': self.policy_svhn(), 'svhn': self.policy_svhn(),
'reduced_imagenet': self.policy_reduced_imagenet(), 'reduced_imagenet': self.policy_reduced_imagenet(),
'panoptic_deeplab_policy': self.panoptic_deeplab_policy(),
} }
if not policies: if not policies:
...@@ -1888,6 +1889,16 @@ class AutoAugment(ImageAugment): ...@@ -1888,6 +1889,16 @@ class AutoAugment(ImageAugment):
] ]
return policy return policy
@staticmethod
def panoptic_deeplab_policy():
policy = [
[('Sharpness', 0.4, 1.4), ('Brightness', 0.2, 2.0)],
[('Equalize', 0.0, 1.8), ('Contrast', 0.2, 2.0)],
[('Sharpness', 0.2, 1.8), ('Color', 0.2, 1.8)],
[('Solarize', 0.2, 1.4), ('Equalize', 0.6, 1.8)],
[('Sharpness', 0.2, 0.2), ('Equalize', 0.2, 1.4)]]
return policy
@staticmethod @staticmethod
def policy_test(): def policy_test():
"""Autoaugment test policy for debugging.""" """Autoaugment test policy for debugging."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment