def load_features_incident_records(self, incident_records):
original_images = []
actions = []
for block_record_file in incident_records:
with open(block_record_file, 'rb') as f:
block_record = pickle.load(f)
original_images.append(block_record.obs)
actions.append(block_record.action)
features = {}
features['image'] = np.concatenate(
[process_image(image, self.hparams) for image in original_images], axis=0)
features['action'] = np.expand_dims(np.array(actions), axis=1)
return features, original_images
评论列表
文章目录