TUI-NICR / EMSANet

EMSANet: Efficient Multi-Task RGB-D Scene Analysis for Indoor Environments
Apache License 2.0
64 stars 3 forks source link

How to Export train model output to video format #4

Open Seye4 opened 1 year ago

Seye4 commented 1 year ago

How do I export the train model output to video format as shown in your youtube video am referring to this link: https://www.youtube.com/watch?v=-Sopja6UEJg&t=0s

Thanks

danielS91 commented 1 year ago

It is just the input image overlayed with the prediction and some small alpha value. You can easily do it in your pipeline. The function for alpha blending both images is here: https://github.com/TUI-NICR/EMSANet/blob/main/emsanet/visualization.py#L54