We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 441cf4b commit 0953478Copy full SHA for 0953478
PyTorch/Detection/Efficientdet/data/dataset.py
@@ -43,7 +43,7 @@ class CocoDetection(data.Dataset):
43
44
def __init__(self, root, ann_file, config, transform=None):
45
super(CocoDetection, self).__init__()
46
- if isinstance(root, torch._six.string_classes):
+ if isinstance(root, (str, bytes)):
47
root = os.path.expanduser(root)
48
self.root = root
49
self.transform = transform
0 commit comments