o oi{ @sddlmZmZeGdddZeGdddZeGdddZeGdd d ZeGd d d ZeGd d d ZdS)) dataclassfieldc@sJeZdZUdZeed<dZeed<dZeed<dZ e ed<d Z eed <d S) HeatMapRefineCfglocalmode皙?ratiogMbP? valid_thresh num_blocks? overlap_ratioN) __name__ __module__ __qualname__rstr__annotations__rfloatr r intr rrS/home/ubuntu/.local/lib/python3.10/site-packages/kornia/feature/sold2/structures.pyr    rc@s&eZdZUdZeed<dZeed<dS)JunctionRefineCfg num_perturbsg?perturb_intervalN)rrrrrrrrrrrrrs rc@seZdZUdZeed<dZeed<dZeed<dZ e ed<d Z eed <d Z eed <d Z eed<dZeed<dZeed<dZe ed<eedZeed<dZe ed<eedZeed<dS)LineDetectorCfgr detect_thresh@ num_samplesgGz? inlier_threshTuse_candidate_suppressiong@nms_dist_toleranceg333333?heatmap_low_threshrheatmap_high_threshmax_local_patch_radiusg@ lambda_radiususe_heatmap_refinementdefault_factoryheatmap_refine_cfguse_junction_refinementjunction_refine_cfgN)rrrrrrrrr r!boolr"r#r$r&r'r(rrr+r,rr-rrrrr$s           rc@sVeZdZUdZeed<dZeed<dZeed<dZ eed<d Z eed <d Z eed <d S)LineMatcherCfgT cross_checkr min_dist_pts top_k_candidates grid_sizeF line_scoreN) rrrr0r.rrrr3r5r7r8rrrrr/5s     r/c@sJeZdZUdZeed<dZeed<dZeed<dZeed<dZ eed <d S) BackboneCfg input_channelr6depth num_stacksr r1 num_classesN) rrrr;rrr<r>r r?rrrrr9?rr9c@seZdZUeedZeed<dZeed<dZ e ed<dZ eed<d Z e ed <d Ze ed <eedZeed <eedZeed<dS) DetectorCfgr) backbone_cfgFuse_descriptorr2r7Tkeep_border_validgv?detection_threshimax_num_junctionsline_detector_cfgline_matcher_cfgN)rrrrr9rArrBr.r7rrCrDrrErrFr/rGrrrrr@Hs      r@N) dataclassesrrrrrr/r9r@rrrrs