How do I get `evaluate()` to work? about keypoint_rcnn_training_pytorch HOT 2 OPEN

😃 Fixed!

Apparently, the solution on the article was right. But for some reason, when I edit on Colab and save, it doesn't actually get saved. So I had to already have changed coco_eval.py before I even cloned it on my Colab runtime.

I basically did the same thing, but this time I downloaded the training scripts from PyTorch and created my own repository to clone:

1. Download the scripts from pytorch/vision/references/detection and created my own repo storing it
2. Open coco_eval.py
3. Edit the lines somewhere under __init__ for iou_type in iou_types:

   - self.coco_eval[iou_type] = COCOeval(coco_gt, iouType=iou_type)
   + coco_eval = COCOeval(coco_gt, iouType=iou_type)
   + coco_eval.params.kpt_oks_sigmas = np.array([.5, .5, .5, .5]) / 10.0 # 4 keypoints
   + self.coco_eval[iou_type] = coco_eval

4. At the start of the Training Notebook:

   !git clone https://github.com/<username>/<repo-name>.git
   %cd <repo-name>

from keypoint_rcnn_training_pytorch.

Thanks, man. I've been struggling with this error. Apparently, making direct changes to the file in Colab doesn't update it. weird.

from keypoint_rcnn_training_pytorch.