рдХреЗ рд▓рд┐рдП рд╕рдВрджрд░реНрдн ultralytics/data/split_dota.py
рдиреЛрдЯ
рдпрд╣ рдлрд╝рд╛рдЗрд▓ рдпрд╣рд╛рдБ рдЙрдкрд▓рдмреНрдз рд╣реИ https://github.com/ultralytics/ultralytics/рдмреВрдБрдж/рдореБрдЦреНрдп/ultralytics/data/split_dota.py рдХрд╛ рдЙрдкрдпреЛрдЧ рдХрд░реЗрдВред рдпрджрд┐ рдЖрдк рдХреЛрдИ рд╕рдорд╕реНрдпрд╛ рджреЗрдЦрддреЗ рд╣реИрдВ рддреЛ рдХреГрдкрдпрд╛ рдкреБрд▓ рдЕрдиреБрд░реЛрдз рдХрд╛ рдпреЛрдЧрджрд╛рди рдХрд░рдХреЗ рдЗрд╕реЗ рдареАрдХ рдХрд░рдиреЗ рдореЗрдВ рдорджрдж рдХрд░реЗрдВ ЁЯЫая╕Пред ЁЯЩП рдзрдиреНрдпрд╡рд╛рдж !
ultralytics.data.split_dota.bbox_iof(polygon1, bbox2, eps=1e-06)
bbox1 рдФрд░ bbox2 рдХреЗ рдмреАрдЪ iofs рдХреА рдЧрдгрдирд╛ рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
polygon1 |
ndarray
|
рдмрд╣реБрднреБрдЬ рдирд┐рд░реНрджреЗрд╢рд╛рдВрдХ, (рдПрди, 8)ред |
рдЖрд╡рд╢реНрдпрдХ |
bbox2 |
ndarray
|
рдмрд╛рдЙрдВрдбрд┐рдВрдЧ рдмреЙрдХреНрд╕, (рдПрди, 4)ред |
рдЖрд╡рд╢реНрдпрдХ |
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
ultralytics.data.split_dota.load_yolo_dota(data_root, split='train')
DOTA рдбреЗрдЯрд╛рд╕реЗрдЯ рд▓реЛрдб рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
data_root |
str
|
рдбреЗрдЯрд╛ рд░реВрдЯ. |
рдЖрд╡рд╢реНрдпрдХ |
split |
str
|
рд╡рд┐рднрд╛рдЬрд┐рдд рдбреЗрдЯрд╛ рд╕реЗрдЯ, рдЯреНрд░реЗрди рдпрд╛ рд╡реИрд▓ рд╣реЛ рд╕рдХрддрд╛ рд╣реИред |
'train'
|
рдиреЛрдЯреНрд╕
DOTA рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдорд╛рдиреА рдЧрдИ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛: - data_root -рдЫрд╡рд┐рдпрд╛рдВ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓ -рд▓реЗрдмрд▓ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_windows(im_size, crop_sizes=[1024], gaps=[200], im_rate_thr=0.6, eps=0.01)
рдЦрд┐рдбрд╝рдХрд┐рдпреЛрдВ рдХреЗ рдирд┐рд░реНрджреЗрд╢рд╛рдВрдХ рдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
im_size |
tuple
|
рдореВрд▓ рдЫрд╡рд┐ рдЖрдХрд╛рд░, (h, w)ред |
рдЖрд╡рд╢реНрдпрдХ |
crop_sizes |
List(int
|
рдЦрд┐рдбрд╝рдХрд┐рдпреЛрдВ рдХрд╛ рдлрд╕рд▓ рдЖрдХрд╛рд░ред |
[1024]
|
gaps |
List(int
|
рдлрд╕рд▓реЛрдВ рдХреЗ рдмреАрдЪ рдЕрдВрддрд░ред |
[200]
|
im_rate_thr |
float
|
рдЫрд╡рд┐ рдПрд░реЗрд╕ рджреНрд╡рд╛рд░рд╛ рд╡рд┐рднрд╛рдЬрд┐рдд рдЦрд┐рдбрд╝рдХрд┐рдпреЛрдВ рдХреНрд╖реЗрддреНрд░реЛрдВ рдХреА рджрд╣рд▓реАрдЬред |
0.6
|
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_window_obj(anno, windows, iof_thr=0.7)
рдкреНрд░рддреНрдпреЗрдХ рд╡рд┐рдВрдбреЛ рдХреЗ рд▓рд┐рдП рдСрдмреНрдЬреЗрдХреНрдЯ рдкреНрд░рд╛рдкреНрдд рдХрд░реЗрдВред
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
ultralytics.data.split_dota.crop_and_save(anno, windows, window_objs, im_dir, lb_dir)
рдЫрд╡рд┐рдпреЛрдВ рдХреЛ рдХреНрд░реЙрдк рдХрд░реЗрдВ рдФрд░ рдирдП рд▓реЗрдмрд▓ рд╕рд╣реЗрдЬреЗрдВред
рдкреИрд░рд╛рдореАрдЯрд░:
рдирд╛рдо | рдкреНрд░рдХрд╛рд░ | рдпрд╛ рдХрд╝рд┐рд╕реНтАНрдо | рдЪреВрдХ |
---|---|---|---|
anno |
dict
|
рдПрдиреЛрдЯреЗрд╢рди рдбрд┐рдХреНрдЯ, рд╕рд╣рд┐рдд |
рдЖрд╡рд╢реНрдпрдХ |
windows |
list
|
рд╡рд┐рдВрдбреЛрдЬрд╝ рдирд┐рд░реНрджреЗрд╢рд╛рдВрдХ рдХреА рдПрдХ рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
window_objs |
list
|
рдкреНрд░рддреНрдпреЗрдХ рд╡рд┐рдВрдбреЛ рдХреЗ рдЕрдВрджрд░ рд▓реЗрдмрд▓ рдХреА рдПрдХ рд╕реВрдЪреАред |
рдЖрд╡рд╢реНрдпрдХ |
im_dir |
str
|
рдЫрд╡рд┐рдпреЛрдВ рдХрд╛ рдЖрдЙрдЯрдкреБрдЯ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдкрдеред |
рдЖрд╡рд╢реНрдпрдХ |
lb_dir |
str
|
рд▓реЗрдмрд▓ рдХрд╛ рдЖрдЙрдЯрдкреБрдЯ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рдкрдеред |
рдЖрд╡рд╢реНрдпрдХ |
рдиреЛрдЯреНрд╕
DOTA рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдорд╛рдиреА рдЧрдИ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛: - data_root -рдЫрд╡рд┐рдпрд╛рдВ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓ -рд▓реЗрдмрд▓ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_images_and_labels(data_root, save_dir, split='train', crop_sizes=[1024], gaps=[200])
рдЫрд╡рд┐рдпреЛрдВ рдФрд░ рд▓реЗрдмрд▓ рджреЛрдиреЛрдВ рдХреЛ рд╡рд┐рднрд╛рдЬрд┐рдд рдХрд░реЗрдВред
рдиреЛрдЯреНрд╕
DOTA рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдорд╛рдиреА рдЧрдИ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛: - data_root -рдЫрд╡рд┐рдпрд╛рдВ -рдЪреАрд░рдирд╛ -рд▓реЗрдмрд▓ -рдЪреАрд░рдирд╛ рдФрд░ рдЖрдЙрдЯрдкреБрдЯ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛ рд╣реИ: - save_dir -рдЫрд╡рд┐рдпрд╛рдВ -рдЪреАрд░рдирд╛ -рд▓реЗрдмрд▓ -рдЪреАрд░рдирд╛
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_trainval(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
рд╕реНрдкреНрд▓рд┐рдЯ рдЯреНрд░реЗрди рдФрд░ рдбреАрдУрдЯреАрдП рдХрд╛ рд╡реИрд▓ рд╕реЗрдЯред
рдиреЛрдЯреНрд╕
DOTA рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдорд╛рдиреА рдЧрдИ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛: - data_root -рдЫрд╡рд┐рдпрд╛рдВ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓ -рд▓реЗрдмрд▓ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓ рдФрд░ рдЖрдЙрдЯрдкреБрдЯ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛ рд╣реИ: - save_dir -рдЫрд╡рд┐рдпрд╛рдВ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓ -рд▓реЗрдмрд▓ -рд░реЗрд▓рдЧрд╛рдбрд╝реА - рд╡реИрд▓
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_test(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
DOTA рдХрд╛ рд╕реНрдкреНрд▓рд┐рдЯ рдЯреЗрд╕реНрдЯ рд╕реЗрдЯ, рд▓реЗрдмрд▓ рдЗрд╕ рд╕реЗрдЯ рдореЗрдВ рд╢рд╛рдорд┐рд▓ рдирд╣реАрдВ рд╣реИрдВред
рдиреЛрдЯреНрд╕
DOTA рдбреЗрдЯрд╛рд╕реЗрдЯ рдХреЗ рд▓рд┐рдП рдорд╛рдиреА рдЧрдИ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛: - data_root -рдЫрд╡рд┐рдпрд╛рдВ -рдкрд░реАрдХреНрд╖рд╛ рдФрд░ рдЖрдЙрдЯрдкреБрдЯ рдирд┐рд░реНрджреЗрд╢рд┐рдХрд╛ рд╕рдВрд░рдЪрдирд╛ рд╣реИ: - save_dir -рдЫрд╡рд┐рдпрд╛рдВ -рдкрд░реАрдХреНрд╖рд╛
рдореЗрдВ рд╕реНрд░реЛрдд рдХреЛрдб ultralytics/data/split_dota.py
2024-01-05 рдмрдирд╛рдпрд╛ рдЧрдпрд╛, рдЕрдкрдбреЗрдЯ рдХрд┐рдпрд╛ рдЧрдпрд╛ 2024-05-08
рд▓реЗрдЦрдХ: рдмреБрд░рд╣рд╛рди-рдХреНрдпреВ (1), рдЧреНрд▓реЗрди-рдЬреЛрдЪрд░ (2)