GitLab

PiperOrigin-RevId: 361897677

PiperOrigin-RevId: 360482921

PiperOrigin-RevId: 359406239

PiperOrigin-RevId: 359344675

PiperOrigin-RevId: 358432885

to export CenterNet model by passing in flags  without changing the pipeline configuration.

PiperOrigin-RevId: 358432613

MobileNetV2Fpn feature extractor.

PiperOrigin-RevId: 358212798

PiperOrigin-RevId: 358070404

PiperOrigin-RevId: 355908695

PiperOrigin-RevId: 355732989

PiperOrigin-RevId: 355320914

keypoint depth.

Updated model_lib files to populate the keypoint depth information.

PiperOrigin-RevId: 355261727

PiperOrigin-RevId: 354590046

PiperOrigin-RevId: 354429469

properly changed if the keep_aspect_ratio_resizer was used in conjunction with
pad_to_max_dimension.

PiperOrigin-RevId: 354336467

depth and weight fields.

PiperOrigin-RevId: 353975078

PiperOrigin-RevId: 353788624

Updated the tf_example decoder and decoder builders to parse the keypoint depth features from the input tf.Example.

PiperOrigin-RevId: 353305509

PiperOrigin-RevId: 352873888

PiperOrigin-RevId: 352851428

estimation. It is designed specifically to run the model in the browser
environment using tf.js.

PiperOrigin-RevId: 352609409

PiperOrigin-RevId: 352601466

PiperOrigin-RevId: 351619683

PiperOrigin-RevId: 351610384

FPN network.

PiperOrigin-RevId: 351255218

for object detection and keypoint estimation tasks.

PiperOrigin-RevId: 350640225

PiperOrigin-RevId: 347837387

by default NMS is off.

PiperOrigin-RevId: 347613472

Context R-CNN Updates: Added capabilities to use multi-headed or multi-layered attention, to place the attention heads pre- or post-second stage feature extraction, and to work with embedded features in the context feature back from pre- or post-second stage feature extraction. Added an option for RPN feature map crops to be piped through to model outputs.

PiperOrigin-RevId: 345777219

DEFAULT_VALUE_OK=setting prefetch batches default to 2 as using -1 can cause memory issues for some models such as context r-cnn that use sequence examples.

PiperOrigin-RevId: 345682268

PiperOrigin-RevId: 345620862

PiperOrigin-RevId: 345586686

PiperOrigin-RevId: 345586199

Updating Context R-CNN dataset tools to fix a bug in context feature bank building logic when the number of context examples in the time horizon is greater than the specified maximum, and adding capabilities to track and save the number of embeddings stored per image.

PiperOrigin-RevId: 345548385

added a new operating mode for single-instance prediction.

1) Refactored the _postprocess_keypoints_for_class_and_image function such that
it can be reused by single/multi class keypoint tasks.
2) Removed the "mod" operator to make the model compatible with WASM.

PiperOrigin-RevId: 345468250

Avoid  explicit tf.unstack in `convert_strided_predictions_to_normalized_boxes`. This enables exporting a saved model with dynamic batch size.

PiperOrigin-RevId: 345055072

PiperOrigin-RevId: 344743049

PiperOrigin-RevId: 344310944

Fix the score_converter issue, which should apply on the dimension of num_class_slots per anchor separately, not on all anchors per location together (e.g. think of softmax).

PiperOrigin-RevId: 344143469