#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(