Datasets
Models
Downloads: 1
129 lines (129 with data), 13.9 kB
B
%öYfyC Ń
@ sb d Z ddlZddlZddlZddlZddlZddlmZ ddl
m m Z ddl
mZ ddlmZmZmZmZmZmZ ddlmZ dZdZG dd ä d âZeed
fd d äZeefd
däZ d<ddäZ d=ddäZ ddä Z d>ddäZ d?d!d"äZ!d@d#d$äZ"dAd%d&äZ#d'd(ä Z$dBd,d-äZ%dd.d/d0dd d1eed
f
d2d3äZ&dCd4d5äZ'G d6d7ä d7âZ(G d8d9ä d9âZ)G d:d;ä d;âZ*dS )Dz
Image augmentation functions
Ú N)┌Image)┌LOGGER┌
check_version┌colorstr┌resample_segments┌ segment2box┌
xywhn2xyxy)┌bbox_ioa)g
Îúp=
▀?g╔vżč/Ţ?gľCőlšű┘?)g Zd;▀O═?gyÚ&1Č╠?g═╠╠╠╠╠╠?c @ s e Zd ZdddäZd ddäZdS )
┌AlbumentationsÚÇ c C s d | _ tdâ}y┤dd l}t|jdddŹ |j||dddd Ź|jd
d Ź|jd
d Ź|jd
d Ź|j d
d Ź|j
dd Ź|j dd Ź|j d dd
Źg}|j
||jddgdŹdŹ| _ tá|dáddä |D âí í W nH tk
rÍ Y n6 tk
Ér
} ztá|Ť |Ť Łí W d d }~X Y nX d S )Nzalbumentations: r z1.0.3T)┌hard)gÜÖÖÖÖÖÚ?g ?)g═╠╠╠╠╠ý?g├§(\Ć┬˝?g )┌height┌width┌scale┌ratio┌pg{«Gßzä?)r ÚK )┌
quality_lowerr ┌yolo┌ class_labels)┌format┌ label_fields)┌ bbox_paramsz, c s s" | ]}|j r|Ť ád díV qdS )zalways_apply=False, ┌ N)r ┌replace)┌.0┌xę r ˙R/home/iml/Desktop/Talha/yolov5_P2_ROI_HCM_Micro_1000_Github/utils/augmentations.py˙ <genexpr>, s z*Albumentations.__init__.<locals>.<genexpr>)┌ transformr ┌albumentationsr ┌ __version__┌RandomResizedCrop┌Blur┌
MedianBlur┌ToGray┌CLAHE┌RandomBrightnessContrast┌ RandomGamma┌ImageCompression┌Compose┌
BboxParamsr ┌info┌join┌ ImportError┌ Exception)┌self┌size┌prefix┌A┌T┌er r r ┌__init__ s&
"zAlbumentations.__init__š ?c C sr | j rjtáí |k rj| j ||d d ůdd ůf |d d ůdf dŹ}|d táddä t|d |d âD âí }}||fS ) NÚ r )┌image┌bboxesr r: c S s g | ]\}}|f|ĽĹqS r r )r ┌c┌br r r ˙
<listcomp>5 s z+Albumentations.__call__.<locals>.<listcomp>r r; )r ┌random┌np┌array┌zip)r1 ┌im┌labelsr ┌newr r r ┌__call__2 s ,,zAlbumentations.__call__N)r )r8 )┌__name__┌
__module__┌ __qualname__r7 rF r r r r r
s
r
Fc C s t j| |||dŹS )N)┌inplace)┌TF┌ normalize)r ┌mean┌stdrJ r r r rL 9 s rL c C sB x<t dâD ]0}| d d ů|f || || | d d ů|f< q
W | S )NÚ )┌range)r rM rN ┌ir r r ┌ denormalize> s 0rR š Ó?c C sŔ |s |s |rńt jádddí|||g d }tátá| tjíí\}}}| j}t jdd|jdŹ} | |d d á |í}
t á
| |d ddíá |í} t á
| |d ddíá |í} tá tá ||
ítá || ítá || ífí}
tj|
tj
| d
Ź d S ) NÚ r9 rO r Ú )┌dtypeÚ┤ Ú Ú )┌dst)r@ r? ┌uniform┌cv2┌split┌cvtColor┌
COLOR_BGR2HSVrV ┌arange┌astype┌clip┌merge┌LUT┌
COLOR_HSV2BGR)rC ┌hgain┌sgain┌vgain┌r┌hue┌sat┌valrV r ┌lut_hue┌lut_sat┌lut_val┌im_hsvr r r ┌ augment_hsvE s (rq Tc C sá t á| |rt jnt jí}|rZt jdddŹ}|á|d d ůd d ůdf í|d d ůd d ůdf< n.t á|d d ůd d ůdf í|d d ůd d ůdf< t á||rśt jnt jíS )Ng @)Ú rr )Z clipLimitZ tileGridSizer ) r\ r^ ┌
COLOR_BGR2YUV┌
COLOR_RGB2YUV┌ createCLAHE┌apply┌ equalizeHist┌
COLOR_YUV2BGR┌
COLOR_YUV2RGB)rC Zclahe┌bgrZyuvr< r r r ┌
hist_equalizeU s 0.r{ c C s* | j d dů \}}|d d ůdd ůf átí}|j\}}}}|| || d } xĎ| áí d t| jd âů D ]┤}
||
\} } }
}|| |
| }}ttád|| íâttád|| íâ }}|||| || g\}}}}| | |ů| |
ůf | ||ů||ůf< t j
|||
df ||||ggddŹ}qjW | |fS )NrY r9 g Ó?r )┌axis) ┌shapera ┌intr5 ┌argsort┌roundr2 r? r[ r@ ┌append)rC rD ┌h┌w┌boxes┌x1┌y1┌x2┌y2┌srQ ┌x1b┌y1b┌x2b┌y2b┌bh┌bw┌yc┌xc┌x1a┌y1a┌x2a┌y2ar r r ┌ replicate` s * $(rľ ęiÇ iÇ ęÚr rÖ rÖ Ú c C só | j d dů }t|târ ||f}t|d |d |d |d â}|sPt|dâ}||f} tt|d | ââtt|d | ââf}
|d |
d |d |
d } } |r┬tá| |ítá| |í } } n>|Ér d\} } |d |d f}
|d |d |d |d f} | d } | d } |d d dů |
kÉr6tj| |
tj dŹ} tt| d ââtt| d ââ }
}tt| d ââtt| d ââ }}tj
| |
|||tj |d Ź} | | | | ffS )
NrY r r9 g ?)g g rT )┌
interpolationgÜÖÖÖÖÖ╣?)┌value) r} ┌
isinstancer~ ┌minrÇ r@ ┌modr\ ┌resize┌ INTER_LINEAR┌copyMakeBorder┌BORDER_CONSTANT)rC ┌ new_shape┌color┌auto┌ scaleFill┌scaleup┌strider} ri r ┌ new_unpad┌dw┌dh┌top┌bottom┌left┌rightr r r ┌ letterboxq s.
"
(" ""r▒ r Ú
šÜÖÖÖÖÖ╣?š ęr r c
C s~ | j d |d d } | j d |d d }
tádí} | j d d | d< | j d d | d< tádí} tá| |í| d< tá| |í| d< tádí}
tá| |í}tád| d| í}tj|d |d
Ź|
d dů< tádí}tátá| |ítj d í|d < tátá| |ítj d í|d
< tádí}tád| d| í|
|d< tád| d| í| |d< || |
| | }|d dkÉsĄ|d dkÉsĄ|tádíká
í ÉrÓ|Ér┬tj | ||
| fddŹ} n tj | |d dů |
| fddŹ} t
|â}|Érvt
ddä |D ââÉo t
|â|k}tá|dfí}|Ér╚t|â}xśt|âD ]î\}}tát
|âdfí}||d d ůd důf< ||j }|ÉrÜ|d d ůd důf |d d ůddůf n|d d ůd důf }t||
| â||< Éq4W Én^tá|d dfí}|d d ůddddddddgf á|d dí|d d ůd důf< ||j }|ÉrN|d d ůd důf |d d ůddůf n|d d ůd důf á|dí}|d d ůddddgf }|d d ůddddgf }tá|ádí|ádí|ádí|ádífíád|íj}|d d ůddgf ád|
í|d d ůddgf< |d d ůddgf ád| í|d d ůddgf< t|d d ůddůf j| |j|ÉrNdnddŹ}|| }|| |d d ůddůf< | |fS )Nr rY r9 rO )r rY )r9 rY )rY r )rY r9 )r r )┌angle┌centerr rW )r r9 )r9 r g Ó?)rÖ rÖ rÖ )┌dsize┌ borderValuec s s | ]}|á í V qd S )N)┌any)r r r r r r ┘ s z%random_perspective.<locals>.<genexpr>Ú rr Ú Ú Ú g{«Gßzä?gÜÖÖÖÖÖ╣?)┌box1┌box2┌area_thr)r} r@ ┌eyer? r[ r\ ┌getRotationMatrix2D┌math┌tan┌pir║ ┌warpPerspective┌
warpAffine┌len┌zerosr ┌ enumerate┌onesr5 r ┌reshape┌ concatenater× ┌maxrb ┌box_candidates)rC ┌targets┌segments┌degrees┌ translater ┌shear┌ perspective┌borderr
r ┌C┌P┌R┌arë ┌Sr5 ┌M┌n┌ use_segmentsrE rQ ┌segment┌xyr ┌yr r r ┌random_perspectiveŁ s`
""
0
B<
J4,,0rŃ c
C s\ t |â}|ÉrR|ÉrR| j\}}}tá| jtjí}x÷tjt|ât|| âdŹD ]ě} || || }
} ||
d |
d ||
d |
d f} t | |d d ůddůf â}
|
dk á
í rLtá ||
d f| Ľgfdí}|á tá || d d ůddůf | d d ůddůf fdíí t
á||| átjígd d
t
jí qLW t
á| dí}t
á|díátí}|| | |< | ||fS ) N)┌krO rY r9 r╗ rŻ g333333Ë?r rT )r9 r9 r9 )r╔ r} r@ r╩ ┌uint8r? ┌samplerP rÇ r ┌allr╬ rü r\ ┌ drawContoursra ┌int32┌FILLED┌flip┌bool)rC rD rĎ r rŮ ré râ r< ┌im_new┌j┌lrë ┌box┌ioa┌resultrQ r r r ┌
copy_pasteř s $ :& rˇ c C sV t á í |k ÉrR| jd dů \}}dgd dgd dgd dgd d gd
}É xŘ|D ]˘}t ádt|| âí}t ádt|| âí}td t ád |í|d â} td t ád |í|d â}
t|| | â} t||
| â} d d
ä tdâD â| |
| ů| | ůf< t|ârZ|dkrZtj | |
| | gtj
dŹ}
t |
t |d d ůddůf ||ââ}||dk }qZW |S )NrY g Ó?r9 g đ?g └?r╗ g ░?rr g á?Ú r c S s g | ]}t ád díĹqS )Ú@ Ú┐ )r? ┌randint)r ┌_r r r r> # s zcutout.<locals>.<listcomp>rO gŞ ůŰQŞ×?)rV rŻ g333333Ń?)
r? r} r¸ r~ r¤ r× rP r╔ r@ rA ┌float32r r )rC rD r ré râ ┌scalesrë ┌mask_h┌mask_w┌xmin┌ymin┌xmax┌ymaxr r˝ r r r ┌cutout s 2 ""r c C sB t jáddí}| | |d| át jí} t á||fdí}| |fS )Ng @@r9 r )r@ r? ┌betara rň r╬ )rC rD ┌im2┌labels2ri r r r ┌mixup. s r rY Úd š╝ëěŚ▓Ďť<c C sÉ | d | d | d | d }}|d |d |d |d }} t á|| | | || í}
||k| |k@ || || | |k@ |
|k @ S )NrY r rO r9 )r@ ┌maximum) r┐ r└ ┌wh_thr┌ar_thrr┴ ┌eps┌w1┌h1┌w2┌h2┌arr r r rđ 6 s "" rđ ÚÓ )g{«Gßz┤?g ?)g Ŕ?gUUUUUU§?gÜÖÖÖÖÖ┘?c
C sv t dâ}
Éy dd l} ddlm} t| jdddŹ | rż| j||||dŹg}
| r^tá|
Ť dŁí qě|dkrx|
| j |d Źg7 }
|dkrĺ|
| j
|d Źg7 }
|dkrět |âfd
}|
| j |d×Ä g7 }
n| j
|d Ź| j||d Źg}
|
| j||d
Ź| â g7 }
tá|
dáddä |
D âí í | á|
íS tk
Ér< tá|
Ť dŁí Y n6 tk
Érp } ztá|
Ť |Ť Łí W d d }~X Y nX d S )Nzalbumentations: r )┌
ToTensorV2z1.0.3T)r )r
r r r z.auto augmentations are currently not supported)r rO )┌max_size)r
r )rM rN z, c s s" | ]}|j r|Ť ád díV qdS )zalways_apply=False, r N)r r )r r r r r r _ s z*classify_albumentations.<locals>.<genexpr>uI ÔÜá´ŞĆ not found, install with `pip install albumentations` (recommended))r )r r! ┌albumentations.pytorchr r r" r# r r- ┌HorizontalFlip┌ VerticalFlip┌float┌ ColorJitter┌SmallestMaxSize┌
CenterCrop┌ Normalizer. r+ r/ ┌warningr0 )┌augmentr2 r r ┌hflip┌vflipZjitterrM rN ┌auto_augr3 r4 r r5 Z color_jitterr6 r r r ┌classify_albumentations> s0
r! c C s8 t | tâstd| Ť dŁâétát| âtâ tátt ígíS )Nz ERROR: classify_transforms size z# must be integer, not (list, tuple))
rŁ r~ ┌AssertionErrorr5 r+ r ┌ToTensorr ┌
IMAGENET_MEAN┌ IMAGENET_STD)r2 r r r ┌classify_transformsh s r& c s& e Zd Zdç fddä Zddä Zç ZS ) ┌ LetterBoxęiÇ iÇ FrÜ c s8 t â áí t|târ ||fn|\| _| _|| _|| _d S )N)┌superr7 rŁ r~ ré râ rŽ rę )r1 r2 rŽ rę )┌ __class__r r r7 q s
zLetterBox.__init__c sń |j d dů \}}tł j| ł j| â}t|| ât|| â }}ł jr^ç fddä||fD ânł jł j }}t|| d d ât|| d d â } }
tjł jł jdfd|jdŹ} t j
|||ft j dŹ| | | | ů|
|
| ůf< | S ) NrY c 3 s$ | ] }t á|ł j íł j V qd S )N)r─ ┌ceilrę )r r )r1 r r r { s z%LetterBox.__call__.<locals>.<genexpr>gÜÖÖÖÖÖ╣?rO rÖ )rV )rŤ ) r} r× ré râ rÇ rŽ r@ ┌fullrV r\ rá rí ) r1 rC ┌imh┌imwri ré râ ┌hs┌wsrş r» ┌im_outr )r1 r rF w s ** .zLetterBox.__call__)r( FrÜ )rG rH rI r7 rF ┌
__classcell__r r )r* r r' o s r' c s& e Zd Zdç fddä Zddä Zç ZS )r ÚÇ c s, t â áí t|târ ||fn|\| _| _d S )N)r) r7 rŁ r~ ré râ )r1 r2 )r* r r r7 ä s
zCenterCrop.__init__c C sh |j d dů \}}t||â}|| d || d }}tj|||| ů||| ůf | j| jftjdŹS )NrY )rŤ )r} r× r\ rá râ ré rí )r1 rC r- r. ┌mrş r» r r r rF ł s
zCenterCrop.__call__)r3 )rG rH rI r7 rF r2 r r )r* r r é s r c s& e Zd Zdç fddä Zddä Zç ZS )r# Fc s t â áí || _d S )N)r) r7 ┌half)r1 r5 )r* r r r7 Ĺ s
zToTensor.__init__c C sF t á|ádíd d dů í}tá|í}| jr2|áí n|áí }|d }|S )N)rY r r9 rT g Óo@)r@ ┌ascontiguousarray┌ transpose┌torch┌
from_numpyr5 r )r1 rC r r r rF Ľ s
zToTensor.__call__)F)rG rH rI r7 rF r2 r r )r* r r# Ć s r# )rS rS rS )TF)rŚ rś TFTrÜ )r r r▓ r│ r│ r▓ r┤ rÁ )rS )rS )rY r r│ r )r )+┌__doc__r─ r? r\ ┌numpyr@ r8 ┌torchvision.transforms┌
transformsr5 Z!torchvision.transforms.functional┌
functionalrK ┌PILr ┌
utils.generalr r r r r r ┌
utils.metricsr r$ r% r
rL rR rq r{ rľ r▒ rŃ rˇ r r rđ r! r& r' r r# r r r r ┌<module> sZ !
-
X