Просмотр исходного кода

[PyT/EffDet] Replace _six.string_classes with Actual Values

Ao Tang 2 лет назад
Родитель
Сommit
0953478cec
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      PyTorch/Detection/Efficientdet/data/dataset.py

+ 1 - 1
PyTorch/Detection/Efficientdet/data/dataset.py

@@ -43,7 +43,7 @@ class CocoDetection(data.Dataset):
 
 
     def __init__(self, root, ann_file, config, transform=None):
     def __init__(self, root, ann_file, config, transform=None):
         super(CocoDetection, self).__init__()
         super(CocoDetection, self).__init__()
-        if isinstance(root, torch._six.string_classes):
+        if isinstance(root, (str, bytes)):
             root = os.path.expanduser(root)
             root = os.path.expanduser(root)
         self.root = root
         self.root = root
         self.transform = transform
         self.transform = transform