VISTA icon indicating copy to clipboard operation
VISTA copied to clipboard

memory usage

Open lfz opened this issue 9 months ago • 0 comments

https://github.com/Project-MONAI/VISTA/blob/1e7f3cc3ec10c5e9b3bc89b14fdeb2949e45a8e2/monailabel/monaivista/lib/model/vista_point_2pt5/inferer.py#L94

this line use too much memory when image is big

image

image

the definition of variable "pred_volume“ varies in different functions. it is 104 channel prob logits at beginning, but changed to 1 channel segmentation finally. To reduce memory use, it should be claimed as 1 channel segmentation map

lfz avatar Nov 10 '23 06:11 lfz