|h]2ddlZddlmZddlmZddlmZddlmZmZm Z m Z ddl Z ddl Z ddlmZddlmZmZddlmZdd lmZd,d e j.d e j.d ed e j.fdZd-deded e eeeffdZ d.de eefde edfde edfded ed e j.f dZd/deeefde j.ded e e j.fdZ d0deeefde j.de e j.dededed dfd Z d1ded!edede edfde edfd df d"Z! d2ded!ed#ed$ed%e edfd df d&Z" d2ded!ed#ed$ed%e edfd df d'Z#e$d(k(re"d)d*+e#d)d*+yy)3N)glob)ceil)Path)AnyDictListTuple)Image) exif_sizeimg2label_paths)TQDM)check_requirementspolygon1bbox2epsreturnc tdddlm}|jddd}t j |d}t j |d}t j||gd}t j|d d d d dfd d df}t j|d d d dd fd dd f}t j||z dtj} | d | d z} fd tdD\} } } }t j| | | | | || |gdjddd}|Dcgc] }|| }}|Dcgc] }|| }}t j| j}t!t j"| D],}||dj%||dj&||<.t j(|Dcgc]}|j&c}tj*}|d}t j||tj}||z }|j,dk(r|d}|Scc}wcc}wcc}w)a Calculate Intersection over Foreground (IoF) between polygons and bounding boxes. Args: polygon1 (np.ndarray): Polygon coordinates with shape (N, 8). bbox2 (np.ndarray): Bounding boxes with shape (N, 4). eps (float, optional): Small value to prevent division by zero. Returns: (np.ndarray): IoF scores with shape (N, 1) or (N, M) if bbox2 is (M, 4). Notes: Polygon format: [x1, y1, x2, y2, x3, y3, x4, y4]. Bounding box format: [x_min, y_min, x_max, y_max]. zshapely>=2.0.0r)PolygonaxisN.).r).c3,K|] }d|f yw.N).0irs Z/var/www/html/test/engine/venv/lib/python3.12/site-packages/ultralytics/data/split_dota.py zbbox_iof../s@!c1f @sdtyperr)rshapely.geometryrreshapenpminmax concatenatemaximumminimumclipinfrangestackzerosshapezipnonzero intersectionareaarrayfloat32ndim)rrrrlt_pointrb_pointbbox1ltrbwh h_overlapslefttoprightbottompolygon2p sg_polys1 sg_polys2overlapsunionsoutputss ` r!bbox_iofrLs$ '((Aq)HvvhR(HvvhR(H NNHh/b 9E E!T2A2+&c2A2g 7B E!T12+&c12g 7B b!RVV $BFbj(J@uQx@D#ufxxsE3vtVLSUV^^_acdfghH%-..I.%-..I.xx (()H "**Z( )J!o229QrU3CDII J XXy1!qvv1 DF I F WWVS"&& )FG||q)$ N/.2s;I II data_rootsplitc|dvs Jd|dt|dz |z }|js Jd|dttt|dz |z dz }t |}g}t ||D]\}}t tj|\}} t|d 5} | jjjD cgc]} t| s| j } } tj| tj } d d d |j#t%| |f | |Scc} w#1swY1xYw) a1 Load DOTA dataset annotations and image information. Args: data_root (str): Data root directory. split (str, optional): The split data set, could be 'train' or 'val'. Returns: (List[Dict[str, Any]]): List of annotation dictionaries containing image information. Notes: The directory structure assumed for the DOTA dataset: - data_root - images - train - val - labels - train - val >valtrainz$Split must be 'train' or 'val', not .images Can't find , please check your data root.*utf-8encodingr#N)ori_sizelabelfilepath)rexistsrstrr r3r r openreadstrip splitlineslenrNr'r7r8appenddict) rMrNim_dirim_fileslb_filesannosim_filelb_filewhfxlbs r!load_yolo_dotarqAsJ* $ $U(LUGST&UU $ )_x '% /F ==?Pk&1OPP?CY(2U:S@ABHx(H E(3HG,-1 'G , 0%&VVX^^%5%@%@%BMc!f!'')MBM"BJJ/B 0  TAq6gFG H LN 0 0s$//EE/E'EEE! im_size crop_sizes.gaps im_rate_thrc |\}}g}t||D];\}} || kDsJd|d| d|| z } ||krdnt||z | z dz} t| D cgc]} | | z } } t| dkDr| d|z|kDr||z | d<||krdnt||z | z dz}t|D cgc]} | | z }} t|dkDr|d|z|kDr||z |d<t j t tj| |tj}||z}|jt j||gd>t j|d}|j}t j|d d dd d fd||d d dd d f<t j|d d dd d fd||d d dd d f<|d d d f|d d dfz |d d d f|d d dfz z}|d d d f|d d dfz |d d d f|d d dfz z}||z }||kDjs$|j}d|t!||z |k<|||kDScc} wcc} w) a6 Get the coordinates of sliding windows for image cropping. Args: im_size (Tuple[int, int]): Original image size, (H, W). crop_sizes (Tuple[int, ...], optional): Crop size of windows. gaps (Tuple[int, ...], optional): Gap between crops. im_rate_thr (float, optional): Threshold of windows areas divided by image areas. eps (float, optional): Epsilon value for math operations. Returns: (np.ndarray): Array of window coordinates with shape (N, 4) where each row is [x_start, y_start, x_stop, y_stop]. zinvalid crop_size gap pair [ ]rrr#rrNr)r3rr/rcr'r7list itertoolsproductint64rdr*copyr-anyr)abs)rrrsrtrurrmrlwindows crop_sizegapstepxnr xsynysstartstop im_in_winsim_areas win_areasim_ratesmax_rates r! get_windowsres( DAqGj$/> 33Q">yk3%q QQ3y.QdA MT+AA+E&F %b *1dQh * * r7Q;2b6I-1]BrFy.QdA MT+AA+E&F %b *1dQh * * r7Q;2b6I-1]BrFi//B78Iy r~~udm!<=!>"nnW1-GJ''*Q1W"5q!34X()C/0 8k) **/+ +s  I"2 I'annoriof_thrc|d\}}|d}t|rl|dddddfxx|zcc<|dddddfxx|zcc<t|ddddf|}tt|Dcgc]}||dd|f|k\c}Stt|Dcgc]'}tjdtj )c}Scc}wcc}w)z3Get objects for each window based on IoF threshold.rZr[Nrr)r r#)rcrLr/r'r1r8) rrrrmrlr[iofsr _s r!get_window_objrs  DAq ME 5z aAg! aAg!ae g.8=c'l8KL1tAqDzW,-LL F F 6FFF( save_dirc `t|dz |z }|jddt|dz |z }|jddt||}t|t ||D]?}t |d||} t || } t|| | t|t|Ay) a Split both images and labels for a given dataset split. Args: data_root (str): Root directory of the dataset. save_dir (str): Directory to save the split dataset. split (str, optional): The split data set, could be 'train' or 'val'. crop_sizes (Tuple[int, ...], optional): Tuple of crop sizes. gaps (Tuple[int, ...], optional): Tuple of gaps between crops. Notes: The directory structure assumed for the DOTA dataset: - data_root - images - split - labels - split and the output directory structure is: - save_dir - images - split - labels - split rSTparentsexist_oklabels)rNtotaldescrZN) rmkdirrqr rcrrrr^) rMrrNrsrtrfrrirrrs r!split_images_and_labelsrs>(^h & .F LLL- (^h & .F LLL- 9E 2EU#e*59Ld:. DA$T73 dG[#f+s6{KLrrratescgg}}|D]<}|jt||z |jt||z >dD]}t|||||y)am Split train and val sets of DOTA dataset with multiple scaling rates. Args: data_root (str): Root directory of the dataset. save_dir (str): Directory to save the split dataset. crop_size (int, optional): Base crop size. gap (int, optional): Base gap between crops. rates (Tuple[float, ...], optional): Scaling rates for crop_size and gap. Notes: The directory structure assumed for the DOTA dataset: - data_root - images - train - val - labels - train - val and the output directory structure is: - save_dir - images - train - val - labels - train - val >rPrQN)rdrr) rMrrrrrsrtrrNs r!split_trainvalrsj>2J "#i!m,- CaL!""N 8UJMNrc gg}}|D]<}|jt||z |jt||z >t|dz dz }|jddt|dz dz }|j s Jd|dt t |dz } t| t| dD]} ttj| \} } t| | f|| } tj| }t| j}| D]W}|j!\}}}}|d ||z d |d |}|||||f}tj"t ||d z |Yy )a Split test set of DOTA dataset, labels are not included within this set. Args: data_root (str): Root directory of the dataset. save_dir (str): Directory to save the split dataset. crop_size (int, optional): Base crop size. gap (int, optional): Base gap between crops. rates (Tuple[float, ...], optional): Scaling rates for crop_size and gap. Notes: The directory structure assumed for the DOTA dataset: - data_root - images - test and the output directory structure is: - save_dir - images - test rStestTrrTrUrVr)rsrtrrrN)rdrrrr]rr^r rcr r r_rrrrrr)rMrrrrrsrtrrfrgrjrlrmrrrrrrrrrrs r! split_testr.s.2J "#i!m,- CaL!"H~(61H NN4$N/ )_x '& 0F ==?Pk&1OPP?C %&HH FC EG,-1q!f$G ZZ G}!! EF/5}} ,GWffr&7"2!32gYc'KH'&.'&.89H KKH(4'8898 D  E Er__main__DOTAv2z DOTAv2-split)rMr)gư>)rQ)g333333?g{Gz?)gffffff?)T)rQrr)rr)g?)%r{rmathrpathlibrtypingrrrr rnumpyr'PILr ultralytics.data.utilsr r ultralytics.utilsr ultralytics.utils.checksrndarrayfloatrLr^rqrrrboolrrrr__name__rrr!rs)) ="7,rzz,"**,5,BJJ,^!c!#!Dc3h