Searched refs:image_dir (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/mindrecord/tools/ |
| D | imagenet_to_mr.py | 55 def __init__(self, map_file, image_dir, destination, partition_number=1): argument
59 check_filename(image_dir)
60 self.image_dir = image_dir
96 real_path = os.path.join(self.image_dir, label_dict[item])
103 raise PathNotExistsError("not valid image dir in {}".format(self.image_dir))
|
| /third_party/mindspore/mindspore/dataset/datapreprocess/ |
| D | preprocess_imagenet_validate_dataset.py | 41 image_dir = mapping.split(':')
42 old_image_path = os.path.join(validation_dataset_path, image_dir[0])
46 new_image_sub_dir = os.path.join(validation_dataset_path, image_dir[1])
48 new_image_path = os.path.join(new_image_sub_dir, image_dir[0])
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/ |
| D | generate_mindrecord.py | 39 if os.path.isdir(config.image_dir) and os.path.exists(config.anno_path):
40 if not os.path.exists(config.image_dir):
42 raise ValueError(config.image_dir)
|
| /third_party/mindspore/tests/perf_test/mindrecord/imagenet/ |
| D | imagenet_to_tfrecord.py | 27 def get_imagenet_filename_label_pic(map_file, image_dir): argument
48 real_path = os.path.join(image_dir, label_dict[item])
55 raise PathNotExistsError("not valid image dir in {}".format(image_dir))
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | dataset.py | 252 def filter_valid_data(image_dir, anno_path): argument
256 if not os.path.isdir(image_dir):
267 if os.path.isfile(os.path.join(image_dir, file_name)):
273 def data_to_mindrecord_byte_image(image_dir, anno_path, mindrecord_dir, prefix="yolo.mindrecord", f… argument
277 image_files, image_anno_dict = filter_valid_data(image_dir, anno_path)
286 image_path = os.path.join(image_dir, image_name)
|
| /third_party/mindspore/mindspore/lite/tools/converter/config_parser/ |
| D | preprocess_parser.cc | 185 struct dirent *image_dir = readdir(root); in CollectCalibInputs() local
187 while (image_dir != nullptr && count < limited_count) { in CollectCalibInputs()
188 std::string file_name(image_dir->d_name); in CollectCalibInputs()
194 image_dir = readdir(root); in CollectCalibInputs()
|
| /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, argument
156 … yolo_dataset = COCOYoloDataset(root=image_dir, ann_file=anno_path, filter_crowd_anno=filter_crowd,
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/ |
| D | dataset.py | 411 def filter_valid_data(image_dir, anno_path): argument
415 if not os.path.isdir(image_dir):
426 image_path = os.path.join(image_dir, file_name)
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/ |
| D | test_yolov3_darknet53.py | 139 ds, data_size = create_yolo_dataset(image_dir=data_root, anno_path=annFile, is_training=True,
|