wayveai / fiery

PyTorch code for the paper "FIERY: Future Instance Segmentation in Bird's-Eye view from Surround Monocular Cameras"
https://wayve.ai/blog/fiery-future-instance-prediction-birds-eye-view
MIT License
560 stars 85 forks source link

How to Run visualization on custom images using pretrained model inference #50

Open vinay650 opened 1 year ago

vinay650 commented 1 year ago

Can you specify how to convert our Camera input to required format and get the predictions from model.

anthonyhu commented 1 year ago

Hello,

You can check the dataset function that specifies the format of the input batch: https://github.com/wayveai/fiery/blob/master/fiery/data.py#L345