FreeYOLO icon indicating copy to clipboard operation
FreeYOLO copied to clipboard

An error when training."'numpy.float64' object cannot be interpreted as an integer"

Open EQ3000 opened this issue 2 years ago • 1 comments

When running train.py during the process "evaluate", an error is occured.

File "train.py", line 282, in train() File "train.py", line 269, in train path_to_save=path_to_save) File "C:\Users\EQ300\Documents\GitHub\FreeYOLO\engine.py", line 241, in val_one_epoch evaluator.evaluate(model) File "C:\Users\EQ300.conda\envs\freeyolo\lib\site-packages\torch\autograd\grad_mode.py", line 28, in decorate_context return func(*args, **kwargs) File "C:\Users\EQ300\Documents\GitHub\FreeYOLO\evaluator\coco_evaluator.py", line 116, in evaluate cocoEval = COCOeval(self.dataset.coco, cocoDt, annType[1]) File "C:\Users\EQ300.conda\envs\freeyolo\lib\site-packages\pycocotools\cocoeval.py", line 76, in init self.params = Params(iouType=iouType) # parameters File "C:\Users\EQ300.conda\envs\freeyolo\lib\site-packages\pycocotools\cocoeval.py", line 527, in init self.setDetParams() File "C:\Users\EQ300.conda\envs\freeyolo\lib\site-packages\pycocotools\cocoeval.py", line 507, in setDetParams self.iouThrs = np.linspace(.5, 0.95, np.round((0.95 - .5) / .05) + 1, endpoint=True) File "<array_function internals>", line 6, in linspace File "C:\Users\EQ300.conda\envs\freeyolo\lib\site-packages\numpy\core\function_base.py", line 113, in linspace num = operator.index(num)

TypeError: 'numpy.float64' object cannot be interpreted as an integer.

EQ3000 avatar Dec 16 '22 07:12 EQ3000

replace cocoeval.py line 506, 507 to: self.iouThrs = np.linspace(.5, 0.95, 10, endpoint=True) self.recThrs = np.linspace(.0, 1.00, 101, endpoint=True)

EQ3000 avatar Dec 16 '22 08:12 EQ3000