Tài liệu tham khảo cho ultralytics/data/split_dota.py
Ghi
Tệp này có sẵn tại https://github.com/ultralytics/ultralytics/blob/main/ultralytics/dữ liệu/split_dota.py. Nếu bạn phát hiện ra một vấn đề, vui lòng giúp khắc phục nó bằng cách đóng góp Yêu cầu 🛠️ kéo. Cảm ơn bạn 🙏 !
ultralytics.data.split_dota.bbox_iof(polygon1, bbox2, eps=1e-06)
Tính iofs giữa bbox1 và bbox2.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
polygon1 |
ndarray
|
Tọa độ đa giác, (n, 8). |
bắt buộc |
bbox2 |
ndarray
|
Hộp giới hạn, (n ,4). |
bắt buộc |
Mã nguồn trong ultralytics/data/split_dota.py
ultralytics.data.split_dota.load_yolo_dota(data_root, split='train')
Tải tập dữ liệu DOTA.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
data_root |
str
|
Gốc dữ liệu. |
bắt buộc |
split |
str
|
Tập dữ liệu phân tách, có thể là train hoặc val. |
'train'
|
Ghi chú
Cấu trúc thư mục giả định cho tập dữ liệu DOTA: - data_root -Hình ảnh -xe lửa -Val -Nhãn -xe lửa -Val
Mã nguồn trong 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)
Lấy tọa độ của các cửa sổ.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
im_size |
tuple
|
Kích thước ảnh gốc, (h, w). |
bắt buộc |
crop_sizes |
List(int
|
Kích thước cắt của cửa sổ. |
[1024]
|
gaps |
List(int
|
Khoảng cách giữa các loại cây trồng. |
[200]
|
im_rate_thr |
float
|
Ngưỡng của các khu vực cửa sổ chia cho hình ảnh ares. |
0.6
|
Mã nguồn trong ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_window_obj(anno, windows, iof_thr=0.7)
Lấy đối tượng cho mỗi cửa sổ.
Mã nguồn trong ultralytics/data/split_dota.py
ultralytics.data.split_dota.crop_and_save(anno, windows, window_objs, im_dir, lb_dir)
Cắt hình ảnh và lưu nhãn mới.
Thông số:
Tên | Kiểu | Sự miêu tả | Mặc định |
---|---|---|---|
anno |
dict
|
Lệnh chú thích, bao gồm |
bắt buộc |
windows |
list
|
Danh sách tọa độ cửa sổ. |
bắt buộc |
window_objs |
list
|
Một danh sách các nhãn bên trong mỗi cửa sổ. |
bắt buộc |
im_dir |
str
|
Đường dẫn thư mục đầu ra của hình ảnh. |
bắt buộc |
lb_dir |
str
|
Đường dẫn thư mục đầu ra của nhãn. |
bắt buộc |
Ghi chú
Cấu trúc thư mục giả định cho tập dữ liệu DOTA: - data_root -Hình ảnh -xe lửa -Val -Nhãn -xe lửa -Val
Mã nguồn trong ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_images_and_labels(data_root, save_dir, split='train', crop_sizes=[1024], gaps=[200])
Tách cả hình ảnh và nhãn.
Ghi chú
Cấu trúc thư mục giả định cho tập dữ liệu DOTA: - data_root -Hình ảnh -chẻ -Nhãn -chẻ và cấu trúc thư mục đầu ra là: - save_dir -Hình ảnh -chẻ -Nhãn -chẻ
Mã nguồn trong ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_trainval(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Split train và val set của DOTA.
Ghi chú
Cấu trúc thư mục giả định cho tập dữ liệu DOTA: - data_root -Hình ảnh -xe lửa -Val -Nhãn -xe lửa -Val và cấu trúc thư mục đầu ra là: - save_dir -Hình ảnh -xe lửa -Val -Nhãn -xe lửa -Val
Mã nguồn trong ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_test(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Bộ thử nghiệm phân tách của DOTA, nhãn không được bao gồm trong bộ này.
Ghi chú
Cấu trúc thư mục giả định cho tập dữ liệu DOTA: - data_root -Hình ảnh -kiểm tra và cấu trúc thư mục đầu ra là: - save_dir -Hình ảnh -kiểm tra