Release MobileNet V3 models and SSDLite models with MobileNet V3 backbone. (#7678)
* Merged commit includes the following changes:
275131829 by Sergio Guadarrama:
updates mobilenet/README.md to be github compatible adds V2+ reference to mobilenet_v1.md file and fixes invalid markdown
--
274908068 by Sergio Guadarrama:
Opensource MobilenetV3 detection models.
--
274697808 by Sergio Guadarrama:
Fixed cases where tf.TensorShape was constructed with float dimensions
This is a prerequisite for making TensorShape and Dimension more strict
about the types of their arguments.
--
273577462 by Sergio Guadarrama:
Fixing `conv_defs['defaults']` override issue.
--
272801298 by Sergio Guadarrama:
Adds links to trained models for Moblienet V3, adds a version of minimalistic mobilenet-v3 to the definitions.
--
268928503 by Sergio Guadarrama:
Mobilenet v2 with group normalization.
--
263492735 by Sergio Guadarrama:
Internal change
260037126 by Sergio Guadarrama:
Adds an option of using a custom depthwise operation in `expanded_conv`.
--
259997001 by Sergio Guadarrama:
Explicitly mark Python binaries/tests with python_version = "PY2".
--
252697685 by Sergio Guadarrama:
Internal change
251918746 by Sergio Guadarrama:
Internal change
251909704 by Sergio Guadarrama:
Mobilenet V3 backbone implementation.
--
247510236 by Sergio Guadarrama:
Internal change
246196802 by Sergio Guadarrama:
Internal change
246014539 by Sergio Guadarrama:
Internal change
245891435 by Sergio Guadarrama:
Internal change
245834925 by Sergio Guadarrama:
n/a
--
PiperOrigin-RevId: 275131829
* Merged commit includes the following changes:
274959989 by Zhichao Lu:
Update detection model zoo with MobilenetV3 SSD candidates.
--
274908068 by Zhichao Lu:
Opensource MobilenetV3 detection models.
--
274695889 by richardmunoz:
RandomPatchGaussian preprocessing step
This step can be used during model training to randomly apply gaussian noise to a random image patch. Example addition to an Object Detection API pipeline config:
train_config {
...
data_augmentation_options {
random_patch_gaussian {
random_coef: 0.5
min_patch_size: 1
max_patch_size: 250
min_gaussian_stddev: 0.0
max_gaussian_stddev: 1.0
}
}
...
}
--
274257872 by lzc:
Internal change.
--
274114689 by Zhichao Lu:
Pass native_resize flag to other FPN variants.
--
274112308 by lzc:
Internal change.
--
274090763 by richardmunoz:
Util function for getting a patch mask on an image for use with the Object Detection API
--
274069806 by Zhichao Lu:
Adding functions which will help compute predictions and losses for CenterNet.
--
273860828 by lzc:
Internal change.
--
273380069 by richardmunoz:
RandomImageDownscaleToTargetPixels preprocessing step
This step can be used during model training to randomly downscale an image to a random target number of pixels. If the image does not contain more than the target number of pixels, then downscaling is skipped. Example addition to an Object Detection API pipeline config:
train_config {
...
data_augmentation_options {
random_downscale_to_target_pixels {
random_coef: 0.5
min_target_pixels: 300000
max_target_pixels: 500000
}
}
...
}
--
272987602 by Zhichao Lu:
Avoid -inf when empty box list is passed.
--
272525836 by Zhichao Lu:
Cleanup repeated resizing code in meta archs.
--
272458667 by richardmunoz:
RandomJpegQuality preprocessing step
This step can be used during model training to randomly encode the image into a jpeg with a random quality level. Example addition to an Object Detection API pipeline config:
train_config {
...
data_augmentation_options {
random_jpeg_quality {
random_coef: 0.5
min_jpeg_quality: 80
max_jpeg_quality: 100
}
}
...
}
--
271412717 by Zhichao Lu:
Enables TPU training with the V2 eager + tf.function Object Detection training loops.
--
270744153 by Zhichao Lu:
Adding the offset and size target assigners for CenterNet.
--
269916081 by Zhichao Lu:
Include basic installation in Object Detection API tutorial.
Also:
- Use TF2.0
- Use saved_model
--
269376056 by Zhichao Lu:
Fix to variable loading in RetinaNet w/ custom loops. (makes the code rely on the exact name scopes that are generated a little bit less)
--
269256251 by lzc:
Add use_partitioned_nms field to config and update post_prossing_builder to honor that flag when building nms function.
--
268865295 by Zhichao Lu:
Adding functionality for importing and merging back internal state of the metric.
--
268640984 by Zhichao Lu:
Fix computation of gaussian sigma value to create CenterNet heatmap target.
--
267475576 by Zhichao Lu:
Fix for exporter trying to export non-existent exponential moving averages.
--
267286768 by Zhichao Lu:
Update mixed-precision policy.
--
266166879 by Zhichao Lu:
Internal change
265860884 by Zhichao Lu:
Apply floor function to center coordinates when creating heatmap for CenterNet target.
--
265702749 by Zhichao Lu:
Internal change
--
264241949 by ronnyvotel:
Updating Faster R-CNN 'final_anchors' to be in normalized coordinates.
--
264175192 by lzc:
Update model_fn to only read hparams if it is not None.
--
264159328 by Zhichao Lu:
Modify nearest neighbor upsampling to eliminate a multiply operation. For quantized models, the multiply operation gets unnecessarily quantized and reduces accuracy (simple stacking would work in place of the broadcast op which doesn't require quantization). Also removes an unnecessary reshape op.
--
263668306 by Zhichao Lu:
Add the option to use dynamic map_fn for batch NMS
--
263031163 by Zhichao Lu:
Mark outside compilation for NMS as optional.
--
263024916 by Zhichao Lu:
Add an ExperimentalModel meta arch for experimenting with new model types.
--
262655894 by Zhichao Lu:
Add the center heatmap target assigner for CenterNet
--
262431036 by Zhichao Lu:
Adding add_eval_dict to allow for evaluation on model_v2
--
262035351 by ronnyvotel:
Removing any non-Tensor predictions from the third stage of Mask R-CNN.
--
261953416 by Zhichao Lu:
Internal change.
--
261834966 by Zhichao Lu:
Fix the NMS OOM issue on TPU by forcing NMS to run outside of TPU.
--
261775941 by Zhichao Lu:
Make Keras InputLayer compatible with both TF 1.x and TF 2.0.
--
261775633 by Zhichao Lu:
Visualize additional channels with ground-truth bounding boxes.
--
261768117 by lzc:
Internal change.
--
261766773 by ronnyvotel:
Exposing `return_raw_detections_during_predict` in Faster R-CNN Proto.
--
260975089 by ronnyvotel:
Moving calculation of batched prediction tensor names after all tensors in prediction dictionary are created.
--
259816913 by ronnyvotel:
Adding raw detection boxes and feature map indices to SSD
--
259791955 by Zhichao Lu:
Added a flag to control the use partitioned_non_max_suppression.
--
259580475 by Zhichao Lu:
Tweak quantization-aware training re-writer to support NasFpn model architecture.
--
259579943 by rathodv:
Add a meta target assigner proto and builders in OD API.
--
259577741 by Zhichao Lu:
Internal change.
--
259366315 by lzc:
Internal change.
--
259344310 by ronnyvotel:
Updating faster rcnn so that raw_detection_boxes from predict() are in normalized coordinates.
--
259338670 by Zhichao Lu:
Add support for use_native_resize_op to more feature extractors. Use dynamic shapes when static shapes are not available.
--
259083543 by ronnyvotel:
Updating/fixing documentation.
--
259078937 by rathodv:
Add prediction fields for tensors returned from detection_model.predict.
--
259044601 by Zhichao Lu:
Add protocol buffer and builders for temperature scaling calibration.
--
259036770 by lzc:
Internal changes.
--
259006223 by ronnyvotel:
Adding detection anchor indices to Faster R-CNN Config. This is useful when one wishes to associate final detections and the anchors (or pre-nms boxes) from which they originated.
--
258872501 by Zhichao Lu:
Run the training pipeline of ssd + resnet_v1_50 + fpn with a checkpoint.
--
258840686 by ronnyvotel:
Adding standard outputs to DetectionModel.predict(). This CL only updates Faster R-CNN. Other meta architectures will be updated in future CLs.
--
258672969 by lzc:
Internal change.
--
258649494 by lzc:
Internal changes.
--
258630321 by ronnyvotel:
Fixing documentation in shape_utils.flatten_dimensions().
--
258468145 by Zhichao Lu:
Add additional output tensors parameter to Postprocess op.
--
258099219 by Zhichao Lu:
Internal changes
--
PiperOrigin-RevId: 274959989
Showing
72.6 KB
79.8 KB
