VhdZddlZddlZddlmZGddejZGddejZGdd ejZGd d ejZ Gd d ejZ dS)zData augmentations. N)nnc*eZdZdZdfd ZdZxZS)Shiftz@ Randomly shift audio in time by up to `shift` samples. Fcdt||_||_dSN)super__init__shiftsame)selfr r __class__s T/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/demucs/augment.pyr zShift.__init__s+   c|\}}}}||jz }|jdkr|js |dd|f}n||jrdn|}t j|j||ddg|j}|d||d}t j||j} | d| |z}|S)Nr.device) sizer trainingr thrandintrexpandarangegather) r wavbatchsourceschannelstimelengthsrcsoffsetsindexess rforwardz Shift.forwards),&w$ " :>>= 7#ww,' I2qq7*TZ%q!1DSZXXX!..WhCC)F3:>>>jjGg$566 r)rF__name__ __module__ __qualname____doc__r r' __classcell__rs@rrrsV       rrceZdZdZdZdS) FlipChannelsz# Flip left-right channels. cr|\}}}}|jr|ddkr}tjd||ddf|j}|ddd|}d|z }tj|d||d|gd}|S)Nrrrdim)rrrrrrcatr)r rrr r!r"leftrights rr'zFlipChannels.forward*s),&w$ = MSXXa[[A--:a%!Q!7 KKKD;;r2r400DHE&#**Q--szz!U/C/CD!LLLC rNr)r*r+r,r'rrr0r0&s-rr0ceZdZdZdZdS)FlipSignz Random sign flip. c|\}}}}|jr6tjd||ddf|jtj}|d|zdz z}|S)Nr2r)rdtype)rrrrrfloat32)r rrr r!r"signss rr'zFlipSign.forward8s`),&w$ = (Jq5'1a"8SUS]^^^EUQ'C rNr8r9rrr;r;4s-rr;c*eZdZdZdfd ZdZxZS)Remixz, Shuffle sources to make new mixes. rcdt||_||_dS)a Shuffle sources within one batch. Each batch is divided into groups of size `group_size` and shuffling is done within each group separatly. This allow to keep the same probability distribution no matter the number of GPUs. Without this grouping, using more GPUs would lead to a higher probability of keeping two sources from the same track together which can impact performance. N)r r proba group_size)r rDrErs rr zRemix.__init__Ds-  $rc |\}}}}|j}|jrtj|jkr|jp|}||zdkrt d|d|||z}||||||}tj tj |||dd|d} | d| ddd||}|||||}|S)Nrz Batch size z! must be divisible by group size rrr3r) rrrrandomrDrE ValueErrorviewrargsortrandrr) r rrstreamsr!r"rrEgroups permutationss rr'z Remix.forwardQs),&w$ = ;V]__tz991EJz!Q&& !cu!c!cWa!c!cdddj(F((6:w$GGC:bgfj'1aX^&_&_&_*+---L**Q 3 3BB$ O OPPC((5'8T::C r)rrBr(r.s@rrArA@sV % % % % % %rrAc&eZdZdfd ZdZxZS)Scale???crt||_||_||_dSr)r r rDminmax)r rDrUrVrs rr zScale.__init__cs2  rc|\}}}}|j}|jrXtj|jkrr]s  BI0     29        ry   BID     BI     r