Fix loadRes() to work with both Python 2 and 3.

waleedka · May 18, 2017 · f83e955 · f83e955
1 parent 336d2a2
commit f83e955
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/PythonAPI/pycocotools/coco.py b/PythonAPI/pycocotools/coco.py
@@ -300,7 +300,12 @@ def loadRes(self, resFile):
 
         print('Loading and preparing results...')
         tic = time.time()
-        if type(resFile) == str or type(resFile) == unicode:
+        # Check result type in a way compatible with Python 2 and 3.
+        try:
+            is_string =  isinstance(resFile, basestring)  # Python 2
+        except NameError:
+            is_string = isinstance(resFile, str)  # Python 3
+        if is_string:
             anns = json.load(open(resFile))
         elif type(resFile) == np.ndarray:
             anns = self.loadNumpyAnnotations(resFile)
@@ -425,4 +430,4 @@ def annToMask(self, ann):
         """
         rle = self.annToRLE(ann)
         m = maskUtils.decode(rle)
-        return m
+        return m