SGT
SGT copied to clipboard
_evaluate_predictions_on_coco( TypeError: _evaluate_predictions_on_coco() got an unexpected keyword argument 'use_fast_impl'
After 5 epoch of training, an error occurs when the model is saved to test the effect.
python projects/SGT/train_net.py --config-file projects/SGT/configs/MOT17/sgt_dla34.yaml --data-dir /root/datasets --num-gpus 1 OUTPUT_DIR /root/sgt_output/mot17_val/dla34_mot17-CH
Traceback (most recent call last):
File "F:\dl\SGT-main\projects\SGT\train_net.py", line 68, in
@HYUNJS
It seems that you are using a different version of detectron2. Please install the detectron2 version stated in the dockerfile (read INSTALL.md of my repo). Otherwise, you should revise the code following the recent version of "_evaluate_predictions_on_coco" function. Please refer to the recent detectron2 using that function.
https://github.com/facebookresearch/detectron2/blob/main/detectron2/evaluation/coco_evaluation.py#L271
@HYUNJS The version I installed for Detectron2 is 0.6 and CUDA is 11.4, but I'm using Windows, not Docker. If I need to modify a file, is it a judgment to remove self._use_fast_impl?