Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 439910408
parent 0d5e7c0a
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
official/vision/configs/retinanet.py
View file @ a5ad25c1
......@@ -126,13 +126,6 @@ class DetectionGenerator(hyperparams.Config):
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
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