Skip to content

GitLab

Commit a10eb09b authored by Fan Yang's avatar Fan Yang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 439910408
parent f25d4ae8
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
official/vision/configs/retinanet.py
View file @ a10eb09b
...@@ -126,13 +126,6 @@ class DetectionGenerator(hyperparams.Config): ...@@ -126,13 +126,6 @@ class DetectionGenerator(hyperparams.Config):
tflite_post_processing: common.TFLitePostProcessingConfig = common.TFLitePostProcessingConfig( tflite_post_processing: common.TFLitePostProcessingConfig = common.TFLitePostProcessingConfig(
) )
max_detections: int = 200
max_classes_per_detection: int = 5
# Regular NMS run in a multi-class fashion and is slow. Setting it to False
# uses class-agnostic NMS, which is faster.
use_regular_nms: bool = False
nms_score_threshold: float = 0.1
@dataclasses.dataclass @dataclasses.dataclass
class RetinaNet(hyperparams.Config): class RetinaNet(hyperparams.Config):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment