@@ -503,8 +503,8 @@ def setDetParams(self):
503
503
self .imgIds = []
504
504
self .catIds = []
505
505
# np.arange causes trouble. the data point on arange is slightly larger than the true value
506
- self .iouThrs = np .linspace (.5 , 0.95 , np .round ((0.95 - .5 ) / .05 ) + 1 , endpoint = True )
507
- self .recThrs = np .linspace (.0 , 1.00 , np .round ((1.00 - .0 ) / .01 ) + 1 , endpoint = True )
506
+ self .iouThrs = np .linspace (.5 , 0.95 , int ( np .round ((0.95 - .5 ) / .05 ) ) + 1 , endpoint = True )
507
+ self .recThrs = np .linspace (.0 , 1.00 , int ( np .round ((1.00 - .0 ) / .01 ) ) + 1 , endpoint = True )
508
508
self .maxDets = [1 , 10 , 100 ]
509
509
self .areaRng = [[0 ** 2 , 1e5 ** 2 ], [0 ** 2 , 32 ** 2 ], [32 ** 2 , 96 ** 2 ], [96 ** 2 , 1e5 ** 2 ]]
510
510
self .areaRngLbl = ['all' , 'small' , 'medium' , 'large' ]
@@ -514,8 +514,8 @@ def setKpParams(self):
514
514
self .imgIds = []
515
515
self .catIds = []
516
516
# np.arange causes trouble. the data point on arange is slightly larger than the true value
517
- self .iouThrs = np .linspace (.5 , 0.95 , np .round ((0.95 - .5 ) / .05 ) + 1 , endpoint = True )
518
- self .recThrs = np .linspace (.0 , 1.00 , np .round ((1.00 - .0 ) / .01 ) + 1 , endpoint = True )
517
+ self .iouThrs = np .linspace (.5 , 0.95 , int ( np .round ((0.95 - .5 ) / .05 ) ) + 1 , endpoint = True )
518
+ self .recThrs = np .linspace (.0 , 1.00 , int ( np .round ((1.00 - .0 ) / .01 ) ) + 1 , endpoint = True )
519
519
self .maxDets = [20 ]
520
520
self .areaRng = [[0 ** 2 , 1e5 ** 2 ], [32 ** 2 , 96 ** 2 ], [96 ** 2 , 1e5 ** 2 ]]
521
521
self .areaRngLbl = ['all' , 'medium' , 'large' ]
@@ -531,4 +531,4 @@ def __init__(self, iouType='segm'):
531
531
raise Exception ('iouType not supported' )
532
532
self .iouType = iouType
533
533
# useSegm is deprecated
534
- self .useSegm = None
534
+ self .useSegm = None
0 commit comments