File size: 745 Bytes
d6def08 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from typing import List, Tuple
from config.config import Config
class EvalConfig(Config):
RPN_PRE_NMS_TOP_N: int = 6000
RPN_POST_NMS_TOP_N: int = 300
@classmethod
def setup(cls, image_min_side: float = None, image_max_side: float = None,
anchor_ratios: List[Tuple[int, int]] = None, anchor_sizes: List[int] = None, pooler_mode: str = None,
rpn_pre_nms_top_n: int = None, rpn_post_nms_top_n: int = None):
super().setup(image_min_side, image_max_side, anchor_ratios, anchor_sizes, pooler_mode)
if rpn_pre_nms_top_n is not None:
cls.RPN_PRE_NMS_TOP_N = rpn_pre_nms_top_n
if rpn_post_nms_top_n is not None:
cls.RPN_POST_NMS_TOP_N = rpn_post_nms_top_n
|