Jelajahi Sumber

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

Ao Tang 2 tahun lalu
induk
melakukan
0953478cec
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  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):
         super(CocoDetection, self).__init__()
-        if isinstance(root, torch._six.string_classes):
+        if isinstance(root, (str, bytes)):
             root = os.path.expanduser(root)
         self.root = root
         self.transform = transform