Files
Track-Anything/dataset/__pycache__/static_dataset.cpython-39.pyc

77 lines
4.8 KiB
Plaintext
Raw Normal View History

2023-04-12 08:24:08 +08:00
a
_{3di<00>@s<>ddlZddlmZddlZddlmZddlmZddlmZddl m
Z
ddl Z ddl mZmZddlmZdd lmZGd
d <0B>d e<04>ZdS) <0C>N)<01>path)<01>Dataset)<01>
transforms)<01>InterpolationMode)<01>Image)<02>im_normalization<6F>im_mean)<01>random_tps_warp)<01>reseedc@s2eZdZdZd dd<05>Zdd<07>Zdd <09>Zd
d <0B>Zd S)<0E>StaticTransformDatasetz<74>
Generate pseudo VOS data by applying random transforms on static images.
Single-object only.
Method 0 - FSS style (class/1.jpg class/1.png)
Method 1 - Others style (XXX.jpg XXX.png)
<20><00>c s<>||_||_g|_|D]<5D>}|\<03>}}|dkr<>t<03><04><01>}|D]H<>t<03>t<05><06><01><00><02>}dd<03>|D<00>} <09><00>fdd<03>| D<00>}
|j<02>|
|<00>q:q|dkr|j<02><07>fdd<03>t<03><04><01>D<00>|<00>qtt |j<02><01>d<07><02>t
<EFBFBD> t
<EFBFBD> dd d d<01>g<01>|_ t
<EFBFBD> t
jd
d d tjtd <0A>t
<EFBFBD>dtj<10>t
jddtd<11>g<03>|_t
<EFBFBD> t
jd
d d tjdd <0A>t
<EFBFBD>dtj<15>t
jdddd<11>g<03>|_t
<EFBFBD> t
<EFBFBD> dd d d <09>t
<EFBFBD>d <09>g<02>|_t
<EFBFBD> t
jddtd<13>t
<EFBFBD><19>g<02>|_t
<EFBFBD> t
jdddd<13>t
<EFBFBD><19>g<02>|_t
<EFBFBD> t
<EFBFBD><1C>tg<02>|_t
<EFBFBD> t
<EFBFBD><1C>g<01>|_dS)NrcSs$g|]}d|dd<02><00><00>vr|<01>qS)<03>jpg<70><67><EFBFBD><EFBFBD><EFBFBD>N)<01>lower<65><02>.0<EFBFBD>im<69>r<00>6/home/gaomingqi/SAM/xmem-sam/dataset/static_dataset.py<70>
<listcomp>$<00>z3StaticTransformDataset.__init__.<locals>.<listcomp>csg|]}t<00><01><01>|<01><03>qSr<00>r<00>joinr<00><02>c<>rootrrr&rr cs g|]}d|vrt<00><01>|<01><02>qS)z.jpgrr)rrrr*rz images found.g<><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?g<><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?<3F>)g<><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?g<><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?<3F>
)<05>degrees<65>scale<6C>shear<61> interpolation<6F>fill<6C><6C>)r$r$T)<02> pad_if_neededr#)g<><67><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>?g<00>?)rr r#) <20>
num_frames<EFBFBD> max_num_obj<62>im_list<73>os<6F>listdirrr<00>extend<6E>print<6E>lenr<00>Compose<73> ColorJitter<65>pair_im_lone_transform<72> RandomAffiner<00>BICUBICr<00>Resize<7A>
RandomCrop<EFBFBD>pair_im_dual_transform<72>NEAREST<53>pair_gt_dual_transform<72>RandomGrayscale<6C>all_im_lone_transform<72>RandomHorizontalFlip<69>all_im_dual_transform<72>all_gt_dual_transform<72>ToTensorr<00>final_im_transform<72>final_gt_transform) <0B>self<6C>
parametersr&r'<00> parameter<65>method<6F>
multiplier<EFBFBD>classes<65>imgsZjpg_listZ
joint_listrrr<00>__init__s^

&<0E> <0E> <0E><08><06><06><02><06>zStaticTransformDataset.__init__c Cs0t<00>|j|<00><01>d<01>}t<00>|j|dd<02>d<00><01>d<04>}tj<05>d<05>}g}g}t|j<08>D]<5D>}t |<04>|<00>
|<02>}|<00> |<08>}t |<04>|<00> |<03>} tj<05>d<05>}
t |
<EFBFBD>|<00> |<08>}|<00>|<08>}t |
<EFBFBD>|<00>| <09>} tj<05><10>dkr<>t|| dd<08>\}} |<00>|<08>}|<00>| <09>} |<05>|<08>|<06>| <09>qVt<15>|d <09>}t<15>|d <09>}||<06><17>fS)
N<EFBFBD>RGBr<00>png<6E>Li<4C><69><EFBFBD>g<1F><>Q<EFBFBD><1E>?g{<14>G<EFBFBD>z<EFBFBD>?)r r)r<00>openr(<00>convert<72>np<6E>random<6F>randint<6E>ranger&r
r;r9r<r5r0r7<00>randr r>r?<00>append<6E>torch<63>stack<63>numpy) r@<00>idxr<00>gtZ sequence_seed<65>images<65>masks<6B>_Zthis_imZthis_gtZ pairwise_seedrrr<00> _get_sampleZs4" 


 





   z"StaticTransformDataset._get_samplecs<>tj<01>|j<03>}|gtjj|<00><04>|d<01><02>}d}tj|jddftjd<03>}t|<03>D]T\}}|<00> |<07>\}} |durp|}n|d| || }|d|| dd<00>dfdk<qL|} t<00>
| d<00>}
|
|
dk}
|
<EFBFBD> <0B>} tj|jddftjd<03>} tjd|jddftjd<03>} t| <0B>D].\}}| |k}|d| |<|d| d|f<<00>qt<00> | d<04>} i<00>|j |<00>d<tdt| <0B><01><02>d<<00>fd d
<EFBFBD>t|j<03>D<00>}t<11>|<10>}|| | |<10>d <0B>}|S) N)<01>sizer$)<01>dtyper rg<00>?<3F>name<6D> num_objectscs g|]}|<01>dkrdnd<02>qS)r_r rr)r<00>i<><01>inforrr<00>rz6StaticTransformDataset.__getitem__.<locals>.<listcomp>)<05>rgb<67>first_frame_gt<67>cls_gt<67>selectorrb)rMrNrOr'<00>__len__<5F>zerosr&<00>int64<36> enumerater[<00>unique<75>tolist<73> expand_dimsr(<00>maxr-rPrS<00> FloatTensor)r@rVZadditional_objects<74>indicesZ merged_imagesZ merged_masksr`Zlist_idrXrY<00>labelsZtarget_objectsrerd<00>lZ this_maskrf<00>datarrar<00> __getitem__<5F>sB   
<02>z"StaticTransformDataset.__getitem__cCs
t|j<01>S)N)r-r()r@rrrrg<00>szStaticTransformDataset.__len__N)r r )<08>__name__<5F>
__module__<EFBFBD> __qualname__<5F>__doc__rGr[rtrgrrrrr s

B&2r )r)rrS<00>torch.utils.data.datasetr<00> torchvisionr<00>torchvision.transformsr<00>PILrrUrM<00>dataset.range_transformrrZ dataset.tpsr Zdataset.reseedr
r rrrr<00><module>s