search

davidtvs / PyTorch-ENet

PyTorch implementation of ENet
MIT License
389 stars 129 forks source link

Can we pass Single image or a video file to the model to test the semantic Segmentation results. #30

Closed prasadpant closed 4 years ago

prasadpant commented 5 years ago

Once the model is trained can we just pass a single image or inputting video file to the model to test the semantic Segmentation results.

davidtvs commented 5 years ago

There's no way to do it from the command-line alone.

You'll have to create a small script to do it. You can reuse the predict function.

prasadpant commented 5 years ago

Will create a script and then call it from command-line with the needed parameters, thanks for your input.