o *j2@sfddlZddlmmZd ddZedkr1ejddZe eZeedd Z e ee e dSdS) NFcCsftjd|ddd}t||gddd}t|d|}|r(tj|ddStj|ddddS)NrT)dimZkeepdimgHz>gP?)r)torchprodcatclamplogFZsoftmax)probkeep_bgZbackground_probZnew_probZlogitsr y/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/models/cv/video_object_segmentation/aggregate.py aggregate s r__main__)rrr)sizeT)r )F) rZtorch.nn.functionalnnZ functionalr r__name__Zrandnr Zsigmoidnewprintr r r r s