Searched defs:create_yolo_dataset (Results 1 – 2 of 2) sorted by relevance
| /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
|