• Francisco Massa's avatar
    [WIP] Add Keypoint R-CNN (#69) · e0a525a0
    Francisco Massa authored
    * [WIP] Keypoints inference on C2 models work
    
    * Training seems to work
    
    Still gives slightly worse results
    
    * e2e training works but gives 3 and 5 mAP less
    
    * Add modification proposed by @ChangErgou
    
    Improves mAP by 1.5 points, to 0.514 and 0.609
    
    * Keypoints reproduce expected results
    
    * Clean coco.py
    
    * Linter + remove unnecessary code
    
    * Merge criteria for empty bboxes in has_valid_annotation
    
    * Remove trailing print
    
    * Add demo support for keypoints
    
    Still need further cleanups and improvements, like adding fields support for the other ops in Keypoints
    
    * More cleanups and misc improvements
    
    * Fixes after rebase
    
    * Add information to the readme
    
    * Fix md formatting
    e0a525a0
Name
Last commit
Last update
.github/ISSUE_TEMPLATE Loading commit data...
configs Loading commit data...
demo Loading commit data...
docker Loading commit data...
maskrcnn_benchmark Loading commit data...
tests Loading commit data...
tools Loading commit data...
.flake8 Loading commit data...
.gitignore Loading commit data...
ABSTRACTIONS.md Loading commit data...
CODE_OF_CONDUCT.md Loading commit data...
CONTRIBUTING.md Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
MODEL_ZOO.md Loading commit data...
README.md Loading commit data...
TROUBLESHOOTING.md Loading commit data...
setup.py Loading commit data...