You need to sign in or sign up before continuing.
Unverified Commit cffac640 authored by Vaibhav Balloli's avatar Vaibhav Balloli Committed by GitHub
Browse files

Fixes EMNIST split and label issues (#2673)



* Add float support to ColorJitter

* Fix byclass EMNIST

* Fix bymerge, balance, letters EMNIST

* Fix whitespace indent

* Revert unrelated file changes

* Revert unrelated file changes

* Removing unnecessary type conversions.

* Removing the transform and adding dummy class instead.
Co-authored-by: default avatarVasileios Vryniotis <vvryniotis@fb.com>
parent 973db145
...@@ -249,13 +249,13 @@ class EMNIST(MNIST): ...@@ -249,13 +249,13 @@ class EMNIST(MNIST):
md5 = "58c8d27c78d21e728a6bc7b3cc06412e" md5 = "58c8d27c78d21e728a6bc7b3cc06412e"
splits = ('byclass', 'bymerge', 'balanced', 'letters', 'digits', 'mnist') splits = ('byclass', 'bymerge', 'balanced', 'letters', 'digits', 'mnist')
# Merged Classes assumes Same structure for both uppercase and lowercase version # Merged Classes assumes Same structure for both uppercase and lowercase version
_merged_classes = set(['C', 'I', 'J', 'K', 'L', 'M', 'O', 'P', 'S', 'U', 'V', 'W', 'X', 'Y', 'Z']) _merged_classes = {'c', 'i', 'j', 'k', 'l', 'm', 'o', 'p', 's', 'u', 'v', 'w', 'x', 'y', 'z'}
_all_classes = set(list(string.digits + string.ascii_letters)) _all_classes = set(string.digits + string.ascii_letters)
classes_split_dict = { classes_split_dict = {
'byclass': list(_all_classes), 'byclass': sorted(list(_all_classes)),
'bymerge': sorted(list(_all_classes - _merged_classes)), 'bymerge': sorted(list(_all_classes - _merged_classes)),
'balanced': sorted(list(_all_classes - _merged_classes)), 'balanced': sorted(list(_all_classes - _merged_classes)),
'letters': list(string.ascii_lowercase), 'letters': ['N/A'] + list(string.ascii_lowercase),
'digits': list(string.digits), 'digits': list(string.digits),
'mnist': list(string.digits), 'mnist': list(string.digits),
} }
......
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