Explorar o código

fix detector evaluate bug

will-jl944 %!s(int64=4) %!d(string=hai) anos
pai
achega
8d51373623
Modificáronse 1 ficheiros con 3 adicións e 0 borrados
  1. 3 0
      dygraph/paddlex/cv/models/detector.py

+ 3 - 0
dygraph/paddlex/cv/models/detector.py

@@ -419,6 +419,9 @@ class BaseDetector(BaseModel):
             if return_details:
             if return_details:
                 return scores, self.eval_details
                 return scores, self.eval_details
             return scores
             return scores
+        if return_details:
+            return None, None
+        return None
 
 
     def predict(self, img_file, transforms=None):
     def predict(self, img_file, transforms=None):
         """
         """