Searched refs:create_yolo_dataset (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/ |
| D | test_yolov3.py | 36 from src.dataset import create_yolo_dataset
116 dataset = create_yolo_dataset(mindrecord_file, repeat_num=1,
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/ |
| D | test_yolov3_darknet53.py | 38 from src.yolo_dataset import create_yolo_dataset
139 ds, data_size = create_yolo_dataset(image_dir=data_root, anno_path=annFile, is_training=True,
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo_dataset.py | 144 def create_yolo_dataset(image_dir, anno_path, batch_size, max_epoch, device_num, rank, function
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | dataset.py | 295 def create_yolo_dataset(mindrecord_dir, batch_size=32, repeat_num=10, device_num=1, rank=0, function
|