Documentation for train_network_no_split_v2

Path: boundary_segmentation/train_network_no_split_v2.py

Functions

register_custom_dataset

No description provided.

print_dataset_stats

No description provided.

train_mask_rcnn

Train Mask R-CNN with optimized hyperparameters

Args: model_type: "mask_rcnn" or "cascade_mask_rcnn" backbone: "resnet50", "resnet101", or "resnext101"

evaluate_on_test

Evaluate the trained model on the test set

run_inference

Run inference on a single image