Index

A

affine_transform() (flat_bug.augmentations.FlatBugRandomPerspective method) annotations_2_contours() (in module flat_bug.coco_utils)

annotations_to_numpy() (in module flat_bug.coco_utils) apply_segments() (in module flat_bug.augmentations)

B

base_nms_() (in module flat_bug.nms) BATCH_SIZE (flat_bug.predictor.Predictor attribute) bbox_intersect() (in module flat_bug.eval_utils) bbox_intersect_area() (in module flat_bug.eval_utils) best_confidence_threshold() (in module flat_bug.eval_utils) bin.fb_eval module

bin.fb_predict module bin.fb_train module build_dataset() (flat_bug.trainers.FlatBugSegmentationTrainer method) build_transforms() (flat_bug.datasets.FlatBugYOLODataset method) (flat_bug.datasets.FlatBugYOLOValidationDataset method)

C

cache_labels() (flat_bug.datasets.FlatBugYOLODataset method) calculate_image_weights() (in module flat_bug.datasets) check_cfg_types() (in module flat_bug.config) check_types() (in module flat_bug.config) chw2hwc_uint8() (in module flat_bug.geometric) clip_boxes() (in module flat_bug.yolo_helpers) compare_groups() (in module flat_bug.eval_utils)

contour_area() (in module flat_bug.coco_utils) contour_bbox() (in module flat_bug.coco_utils) contour_intersection() (in module flat_bug.eval_utils) contour_to_image_coordinates() (flat_bug.predictor.TensorPredictions method) contours (flat_bug.predictor.TensorPredictions property) contours_to_masks() (in module flat_bug.geometric) create_contour_mask() (in module flat_bug.geometric) crop_mask() (in module flat_bug.yolo_helpers)

D

detect_duplicate_boxes() (in module flat_bug.nms)

E

EDGE_CASE_MARGIN (flat_bug.predictor.Predictor attribute) equal_allocate_overlaps() (in module flat_bug.geometric)

equal_spaced_cuts() (in module flat_bug.eval_utils) expand_bottom_right() (in module flat_bug.yolo_helpers) EXPERIMENTAL_NMS_OPTIMIZATION (flat_bug.predictor.Predictor attribute)

F

f1_score() (in module flat_bug.eval_utils) fancy_nms() (in module flat_bug.nms) fb_to_coco() (in module flat_bug.coco_utils) filter_coco() (in module flat_bug.coco_utils) fix_boxes() (flat_bug.predictor.TensorPredictions method) flat_bug.augmentations module flat_bug.coco_utils module flat_bug.config module flat_bug.datasets module flat_bug.eval_utils module

flat_bug.geometric module flat_bug.nms module flat_bug.trainers module flat_bug.yolo_helpers module FlatBugRandomPerspective (class in flat_bug.augmentations) FlatBugSegmentationTrainer (class in flat_bug.trainers) FlatBugYOLODataset (class in flat_bug.datasets) FlatBugYOLOValidationDataset (class in flat_bug.datasets) flip() (flat_bug.predictor.TensorPredictions method) format_cell() (in module flat_bug.eval_utils) format_contour() (in module flat_bug.coco_utils) format_row() (in module flat_bug.eval_utils)

G

generate_block() (in module flat_bug.eval_utils) generate_bootstraps() (in module flat_bug.eval_utils) generate_indices() (in module flat_bug.datasets)

get_dataloader() (flat_bug.trainers.FlatBugSegmentationTrainer method) get_type_def() (in module flat_bug.config) get_validator() (flat_bug.trainers.FlatBugSegmentationTrainer method)

H

HYPERPARAMETERS (flat_bug.predictor.Predictor attribute)

I

iou_boxes() (in module flat_bug.nms) iou_masks() (in module flat_bug.nms)

iou_polygons() (in module flat_bug.nms) IOU_THRESHOLD (flat_bug.predictor.Predictor attribute)

L

linear_interpolate() (in module flat_bug.geometric) load() (flat_bug.predictor.TensorPredictions method)

load_image() (flat_bug.datasets.FlatBugYOLODataset method) low_res_inpaint() (in module flat_bug.augmentations)

M

match_geoms() (in module flat_bug.eval_utils) MAX_MASK_SIZE (flat_bug.predictor.Predictor attribute) merge_tile_results() (in module flat_bug.yolo_helpers) MIN_MAX_OBJ_SIZE (flat_bug.predictor.Predictor attribute) MINIMUM_TILE_OVERLAP (flat_bug.predictor.Predictor attribute) module bin.fb_eval bin.fb_predict bin.fb_train flat_bug.augmentations flat_bug.coco_utils flat_bug.config flat_bug.datasets flat_bug.eval_utils flat_bug.geometric flat_bug.nms flat_bug.trainers flat_bug.yolo_helpers

N

new() (flat_bug.yolo_helpers.ResultsWithTiles method) nms_boxes() (in module flat_bug.nms) nms_masks() (in module flat_bug.nms)

nms_polygons() (in module flat_bug.nms) nms_polygons_() (in module flat_bug.nms) non_max_suppression() (flat_bug.predictor.TensorPredictions method)

O

offset_scale_pad() (flat_bug.predictor.TensorPredictions method)

optimal_threshold_f1() (in module flat_bug.eval_utils)

P

pairwise_contour_intersection() (in module flat_bug.eval_utils) plot() (flat_bug.predictor.TensorPredictions method) plot_heatmap() (in module flat_bug.eval_utils) plot_matches() (in module flat_bug.eval_utils) poly_normals() (in module flat_bug.geometric)

postprocess() (in module flat_bug.yolo_helpers) Predictor (class in flat_bug.predictor) PREFER_POLYGONS (flat_bug.predictor.Predictor attribute) pretty_print_csv() (in module flat_bug.eval_utils) process_mask() (in module flat_bug.yolo_helpers) pyramid_predictions() (flat_bug.predictor.Predictor method)

R

read_cfg() (in module flat_bug.config) resize_mask() (in module flat_bug.geometric)

ResultsWithTiles (class in flat_bug.yolo_helpers) reweight() (in module flat_bug.datasets)

S

save() (flat_bug.predictor.TensorPredictions method) scale_boxes() (in module flat_bug.yolo_helpers) SCORE_THRESHOLD (flat_bug.predictor.Predictor attribute) segment2box() (in module flat_bug.augmentations) serialize() (flat_bug.predictor.TensorPredictions method)

set_hyperparameters() (flat_bug.predictor.Predictor method) setup_model() (flat_bug.trainers.FlatBugSegmentationTrainer method) simplify_contour() (in module flat_bug.geometric) split_annotations() (in module flat_bug.coco_utils) stack_masks() (in module flat_bug.yolo_helpers) subset() (in module flat_bug.datasets)

T

telea_inpaint_polys() (in module flat_bug.augmentations) TensorPredictions (class in flat_bug.predictor)

TILE_SIZE (flat_bug.predictor.Predictor attribute) TIME (flat_bug.predictor.Predictor attribute)

V

validate() (flat_bug.trainers.FlatBugSegmentationTrainer method)

W

write_cfg() (in module flat_bug.config)