• Owen Wang's avatar
    add .npy file handling in evaluator and visualizer · a0ee06f3
    Owen Wang authored
    Summary: Detectron2[Go]'s Visualizer and sem_seg_evaluation now updated with customization entrypoints for how to handle reading semantic seg masks. By default, PIL and PNG images are expected. However, some specific projects' datasets use .npy files and this customization allows providing an alternate Visualizer and evaluation function for reading them.
    
    Reviewed By: newstzpz
    
    Differential Revision: D33434948
    
    fbshipit-source-id: 42af16d6708ffc5b2c03ec8507757313e23c8204
    a0ee06f3
visualization.py 9.26 KB