o ߥi2@sfddlZddlmmZd ddZedkr1ejddZe eZeedd Z e ee e dSdS) NFcCsftjd|ddd}t||gddd}t|d|}|r(tj|ddStj|ddddS)NrT)dimkeepdimgHz>gP?)r)torchprodcatclamplogFsoftmax)probkeep_bgbackground_probnew_problogitsrl/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/cv/video_object_segmentation/aggregate.py aggregate s r__main__)rrr)sizeT)r )F) rtorch.nn.functionalnn functionalr r__name__randnr sigmoidnewprintrrrrs