#if np.any(classes_final[idx][:]== class_id):
#if class_id in set(classes_final[idx]) and idx > split_ind:
if class_id in set(new_classes) and AUGMENT == True:
# skip_augmentation.add(idx)
MINOR_CLASS_FLAG = True
# print('trying to call expand_aug for chip: ', idx)
#im_aug,boxes_aug,classes_aug= aug.expand_aug_random(image, box[idx], classes_final[idx], class_id)
# debug
# added to TF RECORD damaged building only
#im_aug,boxes_aug,classes_aug= aug.expand_aug_random(image, new_coords[new_classes ==1], new_classes[new_classes==1], class_id)
# debug
# augment only sequential chips, not shifted chips
if idx >= sequetial_chip_len:
continue
im_aug, boxes_aug, classes_aug = aug.expand_aug_random(
image, new_coords, new_classes, class_id)
#debug
print('augmentig chip: ', idx)
num_aug = 0
for aug_idx, aug_image in enumerate(im_aug):
# debug
# added to record only damaged buidings
tf_example_aug = tfr.to_tf_example(
aug_image, boxes_aug[aug_idx],
classes_aug[aug_idx])
aug_local_num_class1 = classes_aug[
aug_idx].count(1)
aug_local_num_class2 = classes_aug[
aug_idx].count(2)
#if class_id in set(classes_final[idx]) and idx > split_ind:
if class_id in set(new_classes) and AUGMENT == True:
# skip_augmentation.add(idx)
MINOR_CLASS_FLAG = True
# print('trying to call expand_aug for chip: ', idx)
#im_aug,boxes_aug,classes_aug= aug.expand_aug_random(image, box[idx], classes_final[idx], class_id)
# debug
# added to TF RECORD damaged building only
#im_aug,boxes_aug,classes_aug= aug.expand_aug_random(image, new_coords[new_classes ==1], new_classes[new_classes==1], class_id)
# debug
# augment only sequential chips, not shifted chips
if idx >= sequetial_chip_len:
continue
im_aug, boxes_aug, classes_aug = aug.expand_aug_random(
image, new_coords[new_classes == 1],
new_classes[new_classes == 1], class_id)
#debug
print('augmentig chip: ', idx)
num_aug = 0
for aug_idx, aug_image in enumerate(im_aug):
# debug
# added to record only damaged buidings
tf_example_aug = tfr.to_tf_example(
aug_image, boxes_aug[aug_idx],
classes_aug[aug_idx])
aug_local_num_class1 = classes_aug[aug_idx].count(
1)
aug_local_num_class2 = classes_aug[aug_idx].count(