def __set_chips(self):
directory = self.__filepaths.dir_all
for file in os.listdir(directory):
path = directory + '/' + file
# Only interested in certain files
is_valid = os.path.isfile(path)
is_png = path.endswith(".png")
is_mask = "mask" in path
if not is_valid or not is_png or is_mask:
continue
# Set all Chip variables
car_id = get_sa_car_id(path)
cam_id = get_sa_cam_id(path)
time = None # No timestamp information
misc = None # No miscellaneous information
# Make chip
current_chip = chip.Chip(
path,
car_id,
cam_id,
time,
misc
)
self.chips[path] = current_chip
def __create_chip(self, img_dir, img_name):
# information about the chip resides in the chip's name
splitter = img_name.split("_")
misc = dict()
filepath = img_dir + "/" + img_name
car_id = int(splitter[0])
cam_id = int(utils.get_numeric(splitter[1]))
time = datetime.datetime.fromtimestamp(int(splitter[2]))
misc["binary"] = int(os.path.splitext(splitter[3])[0])
return chip.Chip(filepath, car_id, cam_id, time, misc)
def __create_chip(self, file_info, truth_value):
"""Converts parsing / indexing results into a pelops.datasets.chip.Chip object"""
if truth_value == 0:
self.__noise_seq += 1
car_id = 'unk-{:09d}'.format(self.__noise_seq)
else:
car_id = 'tgt-{:09d}'.format(truth_value)
chip_params = [
file_info['file'], car_id, file_info['meta']['obSetName'],
file_info['meta']['epoch'], file_info['meta']
]
return chip.Chip(*chip_params)
def __create_chip(self, img_dir, img_name):
splitter = img_name.split("/")
misc = dict()
filepath = img_dir + "/" + img_name
car_id = int(splitter[0])
cam_id = None
time = None
misc["color"] = self.__color_map[img_name]
make, model, model_id = self.__model_map[car_id]
misc["make"] = make
misc["model"] = model
misc["model_id"] = model_id
return chip.Chip(filepath, car_id, cam_id, time, misc)
def __create_chip(self, img_dir, img_name):
# information about the chip resides in the chip's name
splitter = img_name.split("_")
misc = {}
filepath = os.path.join(img_dir, img_name)
car_id = int(splitter[0])
cam_id = int(utils.get_numeric(splitter[1]))
time = datetime.datetime.fromtimestamp(int(splitter[2]))
misc["binary"] = int(os.path.splitext(splitter[3])[0])
color, vehicle_type = self.__color_type.get(car_id, (None, None))
misc["color"] = color
misc["vehicle_type"] = vehicle_type
return chip.Chip(filepath, car_id, cam_id, time, misc)
def __set_chips(self):
# identify all the chips, default query to all
name_filepath = {
utils.SetType.ALL.value: self.__filepaths.all_list,
utils.SetType.TEST.value: self.__filepaths.test_list,
utils.SetType.TRAIN.value: self.__filepaths.train_list,
}.get(self.set_type, self.__filepaths.all_list)
# create chip objects based on the names listed in the files
for dg_chip in utils.read_json(name_filepath):
filepath = os.path.normpath(os.path.join(self.dataset_path, dg_chip["filename"]))
car_id = None
cam_id = None
time = None
misc = dg_chip
current_chip = chip.Chip(filepath, car_id, cam_id, time, misc)
self.chips[filepath] = current_chip
