hn2SSKJr SSKrSSKJr SSKJr SSKJr SSKJ r SSK r SSK r SSK Jr SSKJrJr SS KJr SS KJr SSS jjrSSS jjrSSS jjrSSSjjrSS SjjrS!S"SjjrS#S$SjjrS#S$Sjjr\S:Xa\"SSS9 \"SSS9 gg)%) annotationsN)glob)ceil)Path)Any)Image) exif_sizeimg2label_paths)TQDM)check_requirementsc ^[S5 SSKJn URSSS5n[R "USS9n[R "USS9n[R"XE/SS9n[R"US S 2S S S24TS S S245n[R"US S 2S SS 24TS SS 245n[R"X- S[R5n U S U S -n U4S j[S55upp[R"XXXX/SS9RSSS5nUVs/sH nU"U5PM nnUVs/sH nU"U5PM nn[R"U R5n[![R""U 56H-nUUSR%UUS5R&UU'M/ [R("UVs/sHnUR&PM sn[R*S9nUSn[R"UU[R5nUU- nUR,S:XaUSnU$s snfs snfs snf)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).c32># UH nTSU4v M g7f.N).0ibbox2s U/home/james-whalen/.local/lib/python3.13/site-packages/ultralytics/data/split_dota.py bbox_iof..1s@x!c1f xsdtyperr)r shapely.geometryrreshapenpminmax concatenatemaximumminimumclipinfrangestackzerosshapezipnonzero intersectionareaarrayfloat32ndim)polygon1repsrlt_pointrb_pointbbox1ltrbwh h_overlapslefttoprightbottompolygon2p sg_polys1 sg_polys2overlapsunionsoutputss ` rbbox_iofrJs '((Aq)HvvhR(HvvhR(H NNH/b 9E E!T2A2+&c2A2g 7B E!T12+&c12g 7B !RVV $BFbj(J@uQx@DuxxEtLSUV^^_acdfghH%-.XXI.%-.XXI.xx (()H "**Z( )!o229QrU3CDII * XXy1y!qvvy1 DF I F WWVS"&& )FG||q)$ N/.2s/I IIc US;d SUS35e[U5S- U- nUR5(d SUS35e[[[U5S- U- S- 55n[ U5n/n[ X45Hupg[ [R"U55up[USS 9n U R5R5R5V s/sH%n [U 5(dMU R5PM' n n [R"U [R S 9n S S S 5 UR#[%X4W US 95 M U$s sn f!,(df  N4=f) a 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-8encodingrN)ori_sizelabelfilepath)rexistsrstrr r/r ropenreadstrip splitlineslensplitr#r3r4appenddict) data_rootr`im_dirim_fileslb_filesannosim_filelb_filewhfxlbs rload_yolo_dotaroCs;* $ $U(LUGST&UU $ )_x '% /F ==??Pk&1OPP?CY(2U:S@ABHx(H E3G,- 'G ,%&VVX^^%5%@%@%BM%Bc!f)!'')%BBM"BJJ/B-  TA6gFG 4 LN- ,s$1/E E8E %EE E- c UupV/n[X5GH:upX:dSUSU S35eX- n Xh::aSO[Xh- U - S-5n [U 5V s/sHoU -PM n n [U 5S:aU SU-U:aXh- U S'XX::aSO[XX- U - S-5n[U5V s/sHoU -PM nn [U5S:aUSU-U:aXX- US'[R "[ [R"X55[RS9nUU-nUR[R"UU/SS95 GM= [R"USS9nUR5n[R"US S 2SS S 24SU5US S 2SS S 24'[R"US S 2SS S 24SU5US S 2SS S 24'US S 2S 4US S 2S4- US S 2S 4US S 2S4- -nUS S 2S 4US S 2S4- US S 2S 4US S 2S4- -nUU- nUU:R5(d$UR5nSU[!UU- 5U:'UUU:$s sn fs sn f) a 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 [ ]rrrrrNr)r/rr+r_r#r3list itertoolsproductint64rar&copyr)anyr%abs)im_size crop_sizesgaps im_rate_thrr7rkrjwindows crop_sizegapstepxnrxsynysstartstop im_in_winsim_areas win_areasim_ratesmax_rates r get_windowsrgs( DAGj/ Q">yk3%q QQ.QdAMT+AA+E&F %b * 1Qh  * r7Q;2b6I-1]BrF.QdAMT+AA+E&F %b * 1Qh  * r7Q;2b6I-1]BrFi//78Iy r~~udm!<=!0"nnW1-GJ''*Q1W"5q!34X()C/0 8k) **/+ +s I%/I*cUSup4USn[U5(anUSS2SSS24==U-ss'USS2SSS24==U-ss'[USS2SS24U5n[[U55Vs/sHouUSS2U4U:PM sn$[[U55Vs/sH&n[R"S[R S9PM( sn$s snfs snf)z3Get objects for each window based on IoF threshold.rVrWNrr)r r)r_rJr+r#r-r4) annoriof_thrrkrjrWiofsr_s rget_window_objrs  DA ME 5zz aAg! aAg!ae g.8=c'l8KL8K1tAqDzW,-8KLL(^h & .F LLL- (^h & .F LLL- 9 2EU#e*59d:. A$T3 d[#f+s6{K:c//peUH;nUR[X'- 55 UR[X7- 55 M= SHn[XXU5 M g)a  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 >rLrMN)rarr) rcrrrratesr|r}rr`s rsplit_trainvalr sT>2 #im,- CL!" UM"rc //peUH;nUR[X'- 55 UR[X7- 55 M= [U5S- S- nURSSS9 [U5S- S- nUR 5(d SUS35e[ [ US- 55n [U [U 5SS9Hn [[R"U 55up[X4XVS 9n [R"U 5n[U 5RnU HYnUR!5unnnnUS UU- S US U3nUUU2UU24n[R""[ UUS 3- 5U5 M[ M g )aX 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 rOtestTrrPrQrRr)r|r}rrrN)rarrrrYrrZr r_r rr[rrrrrr)rcrrrrr|r}rrdrerhrjrkrrrrrrrrrrs r split_testr0st.2 #im,- CL!H~(61H NN4$N/ )_x '& 0F ==??Pk&1OPP?C %&HH FCG,-qfG ZZ G}!!F/5}} ,GWffr&7"2!32gYc'KH'&.'&.89H KKH(4'8898 D  Dr__main__DOTAv2z DOTAv2-split)rcr)gư>)r6 np.ndarrayrrr7floatreturnr)rM)rcrZr`rZrzlist[dict[str, Any]])g333333?g{Gz?) r{ztuple[int, int]r|tuple[int, ...]r}rr~rr7rrr)gffffff?)rdict[str, Any]rrrrrlist[np.ndarray])T)rrrrrrrdrZrrZrboolrNone)rMrr) rcrZrrZr`rZr|rr}rrr)rr)g?) rcrZrrZrrrrrztuple[float, ...]rr) __future__rrurmathrpathlibrtypingrrnumpyr#PILrultralytics.data.utilsr r ultralytics.utilsr ultralytics.utils.checksr rJrorrrrrr__name__rrrrs# ="7,^!L#*" 2+ 2+2+ 2+ 2+  2+  2+j Q(%) 1L 1L 1L"1L  1L  1L " 1L 1Ln")" (L(L(L (L (L  (L  (LXfl$N$N!$N.1$N?B$NQb$N $NPfl*E*E!*E.1*E?B*EQb*E *EZ zX?N;r