def main():
print('***** Step 2/4: Generate Range-Doppler points *****')
time1 = time.time()
config_path = os.path.join(CARRADA_HOME, 'config.ini')
config = Configurable(config_path).config
warehouse = config['data']['warehouse']
carrada = os.path.join(warehouse, 'Carrada')
with open(os.path.join(carrada, 'data_seq_ref.json'), 'r') as fp:
ref_data = json.load(fp)
with open(os.path.join(carrada, 'validated_seqs.txt')) as fp:
seq_names = fp.readlines()
seq_names = [seq.replace('\n', '') for seq in seq_names]
for seq_name in seq_names:
print('*** Processing sequence {} ***'.format(seq_name))
instances = ref_data[seq_name]['instances']
n_points = 1
time_window = 10
generator = RDPointsGenerator(seq_name, n_points, instances,
time_window)
_, _ = generator.get_rd_points(save_rd_imgs=False,
save_points=True,
save_points_coordinates=False,
save_world_points=True)
print('***** Execution Time for Step 2/4:'
' {} secs. *****'.format(round(time.time() - time1, 2)))
def main():
print('***** Step 3/4: Generate Centroids *****')
time1 = time.time()
config_path = os.path.join(CARRADA_HOME, 'config.ini')
config = Configurable(config_path).config
warehouse = config['data']['warehouse']
carrada = os.path.join(warehouse, 'Carrada')
with open(os.path.join(carrada, 'data_seq_ref.json'), 'r') as fp:
ref_data = json.load(fp)
with open(os.path.join(carrada, 'validated_seqs.txt')) as fp:
seq_names = fp.readlines()
seq_names = [seq.replace('\n', '') for seq in seq_names]
data_types = ['cluster']
for seq_name in seq_names:
print('*** Processing sequence {} ***'.format(seq_name))
ref_ids = ref_data[seq_name]['ref_frame']
instances = ref_data[seq_name]['instances']
min_frame_boundaries = ref_data[seq_name]['min_frame_boundaries']
max_frame_boundaries = ref_data[seq_name]['max_frame_boundaries']
labels = ref_data[seq_name]['labels']
for data_type in data_types:
print(
'===> Generating {} annotations with Jensen-Shannon strategy'.
format(data_type))
CentroidTracking(
seq_name, instances, ref_ids, labels, min_frame_boundaries,
max_frame_boundaries).create_annotations(save_annotations=True,
save_vis_rd=False,
save_vis_doa=False,
data_type=data_type)
print('***** Execution Time for Step 3/4:'
' {} secs. *****'.format(round(time.time() - time1, 2)))
def main():
print('***** Step 4/4: Generate Annotation Files *****')
time1 = time.time()
config_path = os.path.join(CARRADA_HOME, 'config.ini')
config = Configurable(config_path).config
warehouse = config['data']['warehouse']
carrada = os.path.join(warehouse, 'Carrada')
with open(os.path.join(carrada, 'validated_seqs.txt')) as fp:
sequences = fp.readlines()
with open(os.path.join(carrada, 'instance_exceptions.json'), 'r') as fp:
instance_exceptions = json.load(fp)
sequences = [seq.replace('\n', '') for seq in sequences]
annotations_io = get_instance_oriented(sequences, instance_exceptions,
carrada)
annotations_fo = get_frame_oriented(sequences, instance_exceptions,
carrada)
print('***** Execution Time for Step 4/4:'
' {} secs. *****'.format(round(time.time() - time1, 2)))
def main():
print('***** Step 1/4: Generate instances *****')
time1 = time.time()
config_path = os.path.join(CARRADA_HOME, 'config.ini')
config = Configurable(config_path).config
warehouse = config['data']['warehouse']
carrada = os.path.join(warehouse, 'Carrada')
with open(os.path.join(carrada, 'validated_seqs.txt')) as fp:
sequence_names = fp.readlines()
sequence_names = [seq.replace('\n', '') for seq in sequence_names]
for seq_name in sequence_names:
InstanceGenerator().process_sequence(seq_name,
0,
save_boxes=False,
save_masks=False,
save_labels=True,
save_instances_masks=False)
print('***** Execution Time for Step 1/4:'
' {} secs. *****'.format(round(time.time() - time1, 2)))
"""Test function to format data """
import os
import json
import pytest
from carrada_dataset.utils import CARRADA_HOME
from carrada_dataset.utils.configurable import Configurable
from carrada_dataset.scripts.generate_annotation_files import get_instance_oriented, get_frame_oriented
CONFIG_PATH = os.path.join(CARRADA_HOME, 'config.ini')
CONFIG = Configurable(CONFIG_PATH).config
WAREHOUSE = CONFIG['data']['warehouse']
CARRADA = os.path.join(WAREHOUSE, 'Carrada')
SEQ_NAMES = [['2020-02-28-13-14-35'], ['2020-02-28-13-08-51'],
['2019-09-16-13-20-20']]
@pytest.fixture
def get_true_data_instance(request):
with open(os.path.join(CARRADA, 'annotations_instance_oriented.json'),
'r') as fp:
instance_annots = json.load(fp)
return instance_annots[request.param]
@pytest.fixture
def get_true_data_frame(request):
with open(os.path.join(CARRADA, 'annotations_frame_oriented.json'),
'r') as fp:
instance_annots = json.load(fp)
return instance_annots[request.param]
"""Script to set the path to CARRADA in the config.ini file"""
import os
import sys
from carrada_dataset.utils.configurable import Configurable
from carrada_dataset.utils import CARRADA_HOME
if __name__ == '__main__':
path_to_carrada = sys.argv[1]
configurable = Configurable(os.path.join(CARRADA_HOME, 'config.ini'))
configurable.set('data', 'warehouse', path_to_carrada)
with open(os.path.join(CARRADA_HOME, 'config.ini'), 'w') as fp:
configurable.config.write(fp)