Method__init__
:param feature_infos: element 0 means sizes((32, 32), (16, 16), ...), element 1 means channels
model/basic_modules.py:270
Method__init__(self, input_size=(1024, 1024), det_size=(512, 512), reg_size=(144, 48), class_num=68)
model/detection_recognition_pipeline.py:16
Method__init__
:param feature_infos: element 0 means sizes((32, 32), (16, 16), ...), element 1 means channels
model/other_models/shufflenet.py:129
Method__init__
:param feature_infos: element 0 means sizes((32, 32), (16, 16), ...), element 1 means channels
SLPNet_Demo/model/basic_modules.py:270
Method__init__(self, input_size=(1024, 1024), det_size=(512, 512), reg_size=(144, 48), class_num=68)
SLPNet_Demo/model/detection_recognition_pipeline.py:9
Functioncorner2bboxHW
:param corners_maps: tensor size(B, H, W, 8)
:return: size(B, H, W, 4)
module/det_part/GTProcessing.py:142
Functioncorner2bboxHW
:param corners_maps: tensor size(B, H, W, 8)
:return: size(B, H, W, 4)
SLPNet_Demo/utils/GTProcessing.py:142
Methodforward
:param fea_maps: feature maps to Perspective Transform, size(C, H, W)
:param corners_tensor: size(N, 8), N corner pairs number
module/det_part/perspective_transform.py:64
Methodforward
:param fea_maps: feature maps to Perspective Transform, size(C, H, W)
:param corners_tensor: size(N, 8), N corner pairs number
SLPNet_Demo/utils/perspective_transform.py:61
Methodgen_distance_maps
:param out_centers: size(B, H, W, 2)
:param target_centers: size(B, H, W, 2)
:param out_bbox: size(B, H, W, 4)
:p
module/det_part/detection_head.py:321