search

fudan-zvg / Semantic-Segment-Anything

Automated dense category annotation engine that serves as the initial semantic labeling for the Segment Anything dataset (SA-1B).
Apache License 2.0
2.14k stars 137 forks source link

显存占用 #19

Closed houshuaipeng closed 1 year ago

houshuaipeng commented 1 year ago

请问您在推理时使用的显卡显存是多少呢？我使用24G显存的4090推理时爆显存了，请问至少应该使用多少张以及多大显存的显卡呢？

Jiaqi-Chen-00 commented 1 year ago

Hello, we have updated the code and you can pull the latest version. The small version of SSA-engine requires only 12GB of memory, while the base version takes 14GB. (The result is tested on SA-1B dataset) If you wish to use both SSA-engine and SAM together, you will need larger graphics memory. Please refer to the Memory Usage section in the Readme for more detailed information.

houshuaipeng commented 1 year ago

Thank you for your patient answer.