Visualization cost too much time

TRI-ML / realtime_panoptic

Official PyTorch implementation of CVPR 2020 Oral: Real-Time Panoptic Segmentation from Dense Detections

MIT License

115 stars 25 forks source link

Visualization cost too much time #6

Open lucasjinreal opened 4 years ago

lucasjinreal commented 4 years ago

Done with model inference. 0.17570257186889648
Process and visualizing the outputs...
post process: 0.302081823348999
vis cost: 14.560776233673096

Model forward cost 0.17s, postprocess coast 0.3s, while visualization cost 14s...

I am on GTX1080ti, why it was so slow on visualization

jieli352 commented 4 years ago

Hi @jinfagang, we didn't apply any optimization on the visualization code. There are definitely rooms to improve. The cost time also include the I/O of image saving. Contributions are welcome on better visualization as well!

lucasjinreal commented 4 years ago

@jieli352 I think visualize a mask on image and instances should not cost so many time. Root reason is postprocess result data structure not reasonable.