• z55250825's avatar
    Add new parrots extension implementation for all ops (#794) · 48d99025
    z55250825 authored
    * delete all parrots file
    add bbox_overlaps new parrots op impl
    
    * support first new impl parrts op (bbox_overlaps)(success test)
    
    * add box_iou_rotated op, test succeed
    
    * add carafe and carafe_naive op, test succeed (one parrots bug need fix)
    
    * add cc_attention op, test success
    
    * add corner_pool op, test success
    
    * add parrots op deform_conv, test success
    
    * add deform_roi_pool op, test success (but has question)
    
    * add focal loss op, test success (gradcheck)
    
    * add masked_conv2d op, test success
    
    * add modulated_deform_conv op, test success
    
    * add nms and nms_rotated op, test success
    
    * add psamask op, test success
    
    * add roi_align op, test_success
    
    * add roi_pool op, test success
    
    * add sync_bn op, test success
    
    * add tin_shift op, test success
    
    * fix test_deform_roi_pool, add parrots test
    
    * skip test_onnx because parrots does not support onnx
    
    * fix c++ lint
    
    * fix python lint
    
    * fix python lint
    48d99025
setup.py 11.4 KB