Traceback (most recent call last):
File "tran.py", line 336, in <module>
main(args)
File "tran.py", line 282, in main
return trainer.train()
File "/home/zx/detectron2_repo/detectron2/engine/defaults.py", line 484, in train
super().train(self.start_iter, self.max_iter)
File "/home/zx/detectron2_repo/detectron2/engine/train_loop.py", line 156, in train
self.after_step()
File "/home/zx/detectron2_repo/detectron2/engine/train_loop.py", line 190, in after_step
h.after_step()
File "/home/zx/detectron2_repo/detectron2/engine/hooks.py", line 556, in after_step
self._do_eval()
File "/home/zx/detectron2_repo/detectron2/engine/hooks.py", line 529, in _do_eval
results = self._func()
File "/home/zx/detectron2_repo/detectron2/engine/defaults.py", line 453, in test_and_save_results
self._last_eval_results = self.test(self.cfg, self.model)
File "/home/zx/detectron2_repo/detectron2/engine/defaults.py", line 617, in test
results_i = inference_on_dataset(model, data_loader, evaluator)
File "/home/zx/detectron2_repo/detectron2/evaluation/evaluator.py", line 204, in inference_on_dataset
results = evaluator.evaluate()
File "/home/zx/detectron2_repo/detectron2/evaluation/coco_evaluation.py", line 206, in evaluate
self._eval_predictions(predictions, img_ids=img_ids)
File "/home/zx/detectron2_repo/detectron2/evaluation/coco_evaluation.py", line 279, in _eval_predictions
res = self._derive_coco_results(
File "/home/zx/detectron2_repo/detectron2/evaluation/coco_evaluation.py", line 364, in _derive_coco_results
assert len(class_names) == precisions.shape[2]
AssertionError
[zx@415aea10ccba allClassTrainCode]$
改了类别数量后清除缓存,再尝试