OpenDriveLab / MPI

[RSS 2024] Learning Manipulation by Predicting Interaction
https://opendrivelab.com/MPI/
MIT License
78 stars 0 forks source link

how to use pretrained model to extract feature of one single image? #3

Open guangnianyuji opened 1 month ago

guangnianyuji commented 1 month ago

Thanks for this fancy work! And I want to ask how to use pretrained model to extract feature of one single image? Can you give me an example code?

retsuh-bqw commented 1 month ago

Thank you for your interest in our work! We have updated a sample script get_representation.py to facilitate the employment of MPI.