Referência para ultralytics/data/split_dota.py
Nota
Este ficheiro está disponível em https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/data/split_dota .py. Se detectares um problema, por favor ajuda a corrigi-lo contribuindo com um Pull Request 🛠️. Obrigado 🙏!
ultralytics.data.split_dota.bbox_iof(polygon1, bbox2, eps=1e-06)
Calcula os iofs entre bbox1 e bbox2.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
polygon1 |
ndarray
|
Coordenadas do polígono, (n, 8). |
necessário |
bbox2 |
ndarray
|
Caixas de contorno, (n ,4). |
necessário |
Código fonte em ultralytics/data/split_dota.py
ultralytics.data.split_dota.load_yolo_dota(data_root, split='train')
Carrega o conjunto de dados DOTA.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
data_root |
str
|
Raiz dos dados. |
necessário |
split |
str
|
O conjunto de dados dividido, pode ser treinado ou validado. |
'train'
|
Notas
A estrutura de directórios assumida para o conjunto de dados DOTA: - data_root - imagens - treina - val - etiquetas - treina - val
Código fonte em 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)
Obtém as coordenadas das janelas.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
im_size |
tuple
|
Tamanho da imagem original, (h, w). |
necessário |
crop_sizes |
List(int
|
Corta o tamanho das janelas. |
[1024]
|
gaps |
List(int
|
Intervalo entre culturas. |
[200]
|
im_rate_thr |
float
|
Limiar das áreas das janelas dividido pelas áreas da imagem. |
0.6
|
Código fonte em ultralytics/data/split_dota.py
ultralytics.data.split_dota.get_window_obj(anno, windows, iof_thr=0.7)
Obtém objectos para cada janela.
Código fonte em ultralytics/data/split_dota.py
ultralytics.data.split_dota.crop_and_save(anno, windows, window_objs, im_dir, lb_dir)
Recorta imagens e guarda novas etiquetas.
Parâmetros:
Nome | Tipo | Descrição | Predefinição |
---|---|---|---|
anno |
dict
|
Ditado de anotação, incluindo |
necessário |
windows |
list
|
Uma lista de coordenadas de janelas. |
necessário |
window_objs |
list
|
Lista as etiquetas dentro de cada janela. |
necessário |
im_dir |
str
|
O caminho do diretório de saída das imagens. |
necessário |
lb_dir |
str
|
O caminho do diretório de saída das etiquetas. |
necessário |
Notas
A estrutura de directórios assumida para o conjunto de dados DOTA: - data_root - imagens - treina - val - etiquetas - treina - val
Código fonte em ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_images_and_labels(data_root, save_dir, split='train', crop_sizes=[1024], gaps=[200])
Divide as imagens e as etiquetas.
Notas
A estrutura de directórios assumida para o conjunto de dados DOTA: - data_root - imagens - divide - etiquetas - divide e a estrutura do diretório de saída é: - guarda_dir - imagens - divide - etiquetas - divide
Código fonte em ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_trainval(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Separa o comboio e o conjunto de valores do DOTA.
Notas
A estrutura de directórios assumida para o conjunto de dados DOTA: - data_root - imagens - treina - val - etiquetas - treina - val e a estrutura do diretório de saída é: - guarda_director - imagens - train - val - etiquetas - comboio - val
Código fonte em ultralytics/data/split_dota.py
ultralytics.data.split_dota.split_test(data_root, save_dir, crop_size=1024, gap=200, rates=[1.0])
Conjunto de teste dividido do DOTA, as etiquetas não estão incluídas neste conjunto.
Notas
A estrutura de directórios assumida para o conjunto de dados DOTA: - data_root - imagens - teste e a estrutura do diretório de saída é: - guarda_dir - imagens - teste