mirror of
https://github.com/gaomingqi/Track-Anything.git
synced 2025-12-16 16:37:58 +01:00
77 lines
4.8 KiB
Plaintext
77 lines
4.8 KiB
Plaintext
|
a
|
|||
|
|
_{3di <00> @ s<> d dl Z d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dl m
|
|||
|
|
Z
|
|||
|
|
d dl Z d dl
mZmZ d dlmZ d d lmZ G d
|
|||
|
|
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 @ s2 e Zd ZdZd
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> t g<02>| _ t
|
|||
|
|
<EFBFBD> t
|
|||
|
|
<EFBFBD> <1C> g<01>| _ d S )Nr c S s$ g | ] }d |dd<02> <00> <00> v r|<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>c s g | ]}t <00><01><01> |<01><03>qS r <00>r <00>joinr <00><02>c<>rootr r r & r r
c s g | ]}d |v rt <00><01> |<01><02>qS )z.jpgr r )r r r r * r z 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>?)r r r# ) <20>
|
|||
|
|
num_frames<EFBFBD> max_num_obj<62>im_list<73>os<6F>listdirr r <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_listr r r <00>__init__ s^
|
|||
|
|
|
|||
|
|
&<0E> <0E> <0E><08><06><06><02><06>z StaticTransformDataset.__init__c C s0 t <00>| j| <00><01>d<01>}t <00>| j| d d<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> dk r<>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; r9 r< r5 r0 r7 <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_seedr r r <00> _get_sampleZ s4 "
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
z"StaticTransformDataset._get_samplec s<> t j<01>| j<03>}|gt jj| <00><04> |d<01><02>}d }t j| jddft jd<03>}t|<03>D ]T\}}| <00> |<07>\}} |d u rp|}n|d| || }|d || d d <00>df dk< qL|} t <00>
|
|||
|
|
| d <00>}
|
|||
|
|
|
|
|||
|
|
|
|
|||
|
|
dk }
|
|||
|
|
|
|
|||
|
|
<EFBFBD> <0B> } t j| jddft jd<03>} t jd| jddft jd<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
r g <00>?<3F>name<6D> num_objectsc s g | ]}|<01> d k rdnd<02>qS )r_ r
r r )r <00>i<><01>infor r r <00> r z6StaticTransformDataset.__getitem__.<locals>.<listcomp>)<05>rgb<67>first_frame_gt<67>cls_gt<67>selectorrb )rM rN rO r' <00>__len__<5F>zerosr& <00>int64<36> enumerater[ <00>unique<75>tolist<73> expand_dimsr( <00>maxr- rP rS <00> FloatTensor)r@ rV Zadditional_objects<74>indicesZ
merged_imagesZ merged_masksr` Zlist_idrX rY <00>labelsZtarget_objectsre rd <00>lZ this_maskrf <00>datar ra r <00> __getitem__<5F> sB
|
|||
|
|
<02>z"StaticTransformDataset.__getitem__c C s
|
|||
|
|
t | j<01>S )N)r- r( )r@ r r r rg <00> s z StaticTransformDataset.__len__N)r r
)<08>__name__<5F>
|
|||
|
|
__module__<EFBFBD> __qualname__<5F>__doc__rG r[ rt rg r r r r r s
|
|||
|
|
|
|||
|
|
B&2r )r) r rS <00>torch.utils.data.datasetr <00> torchvisionr <00>torchvision.transformsr <00>PILr rU rM <00>dataset.range_transformr r Z dataset.tpsr Zdataset.reseedr
|
|||
|
|
r r r r r <00><module> s
|