• W
    Add mean IOU op. (#10519) · 6fcdb240
    whs 提交于
    * Add mean_iou op.
    
    * Add unitest for mean iou op.
    
    * Add optional collections of confusion matrix and mean_iou.
    
    * Fix cuda kernel.
    
    * Refine code.
    1. Merge computing in GPU to two kernel.
    2. Use wrong array and correct array instead of confusion matrix.
    
    * Add python api and fix cuda kernel.
    
    * Fix comments.
    
    * Small fix.
    
    * Small fix.
    6fcdb240
mean_iou_op.cc 4.3 KB