MCPcopy Create free account

hub / github.com/JackEasson/SLPNet_pytorch / functions

Functions218 in github.com/JackEasson/SLPNet_pytorch

↓ 1 callersFunctionweights_init
(model)
train.py:33
↓ 1 callersFunctionwrap_perspective
:param src_img: 用于wrap的原图 :param corners_list: [tensor, tensor ...],用于corner2bbox函数 :param out_size: 生成的透视后图像尺寸 w*h ?? :return:
module/det_part/PostProcessing.py:191
↓ 1 callersFunctionwrap_perspective
:param src_img: 用于wrap的原图 :param corners_list: [tensor, tensor ...],用于corner2bbox函数 :param out_size: 生成的透视后图像尺寸 w*h ?? :return:
SLPNet_Demo/utils/PostProcessing.py:191
Method__getitem__
:param index: :return: img_tensor size(3, 512, 512); corners_tensor size(N, 8); lp_label_tensor list(obj_num), ea
load_data.py:144
Method__init__
(self, chn_in, chn_out=None, expansion_ratio=1.0)
SSNet_framework.py:11
Method__init__
(self, img_path, txt_path, image_loader=base_image_loader, label_loader=read_txt_info)
load_data.py:128
Method__init__
(self)
module/total_loss.py:8
Method__init__
(self)
module/det_part/perspective_transform.py:61
Method__init__
(self)
module/det_part/detection_head.py:150
Method__init__
(self)
module/det_part/detection_head.py:449
Method__init__
(self, chn_in, chn_out=None, expansion_ratio=1.0)
model/basic_modules.py:53
Method__init__
(self, chn_in, chn_out)
model/basic_modules.py:84
Method__init__
(self, chn_in, chn_out, mode='max', stride=(2, 2))
model/basic_modules.py:106
Method__init__
(self, inp, oup, stride, benchmodel)
model/basic_modules.py:158
Method__init__
:param feature_infos: element 0 means sizes((32, 32), (16, 16), ...), element 1 means channels
model/basic_modules.py:270
Method__init__
:param chn_in: :param size_hw_in: (h, w) or [h, w] :param chn_shrink_ratio:
model/basic_modules.py:315
Method__init__
:param chn_in: a number or a list (tuple) :param size_hw_in: (h, w) or [h, w] :param chn_shrink_ratio:
model/basic_modules.py:342
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 chn_in: :param mode: choose in (0, 1, 2, 3), 0:skip concatenate connection; 1:skip concatenate & add connection;
model/SSNet_modules.py:7
Method__init__
(self, class_num)
model/SSNet_modules.py:175
Method__init__
(self, class_num, det_chns=(32, 80))
model/SSNet_modules.py:253
Method__init__
(self, class_num, det_chns=32)
model/SSNet_modules.py:336
Method__init__
(self, inp, oup, stride, benchmodel)
model/other_models/shufflenet.py:61
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__
(self)
SLPNet_Demo/main.py:21
Method__init__
(self)
SLPNet_Demo/utils/perspective_transform.py:58
Method__init__
(self)
SLPNet_Demo/utils/detection_head.py:126
Method__init__
(self, chn_in, chn_out=None, expansion_ratio=1.0)
SLPNet_Demo/model/basic_modules.py:53
Method__init__
(self, chn_in, chn_out)
SLPNet_Demo/model/basic_modules.py:84
Method__init__
(self, chn_in, chn_out, mode='max', stride=(2, 2))
SLPNet_Demo/model/basic_modules.py:106
Method__init__
(self, inp, oup, stride, benchmodel)
SLPNet_Demo/model/basic_modules.py:158
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__
:param chn_in: :param size_hw_in: (h, w) or [h, w] :param chn_shrink_ratio:
SLPNet_Demo/model/basic_modules.py:315
Method__init__
:param chn_in: a number or a list (tuple) :param size_hw_in: (h, w) or [h, w] :param chn_shrink_ratio:
SLPNet_Demo/model/basic_modules.py:342
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
Method__init__
:param chn_in: :param mode: choose in (0, 1, 2, 3), 0:skip concatenate connection; 1:skip concatenate & add connection;
SLPNet_Demo/model/SSNet_modules.py:8
Method__init__
(self, class_num)
SLPNet_Demo/model/SSNet_modules.py:176
Functionalignment_txt_img
()
data/data_process.py:81
Functionbase_image_loader
返回image的tensor
load_data.py:81
Functionbase_lp_collate
(batch)
load_data.py:177
Functioncoord_trans2real
:param single_reg_map: size(32, 32, 8) :param stage_lvl: scale ratio :param S0: 16 from fovea box :return:
module/det_part/detection_head.py:34
Functioncoord_trans2real
:param single_reg_map: size(32, 32, 8) :param stage_lvl: scale ratio :param S0: 16 from fovea box :return:
SLPNet_Demo/utils/detection_head.py:33
Functioncorner2bbox
:param corners_list: [tensor, tensor ...] :return: list of box, float, not tensor
module/det_part/GTProcessing.py:93
Functioncorner2bbox
:param corners_list: [tensor, tensor ...] :return: list of box, float, not tensor
SLPNet_Demo/utils/GTProcessing.py:93
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
Functioncorner2bboxMulti
:param corners: size(N, 8) :return: size(N, 4)
module/det_part/GTProcessing.py:129
Functioncorner2bboxMulti
:param corners: size(N, 8) :return: size(N, 4)
SLPNet_Demo/utils/GTProcessing.py:129
Functioncorner_scale_extend
:param corner: tensor size(8) :param effective_ratio: :param ignore_ratio: :param max_size :return: as input 'corner'
module/det_part/GTProcessing.py:177
Functioncorner_scale_extend
:param corner: tensor size(8) :param effective_ratio: :param ignore_ratio: :param max_size :return: as input 'corner'
SLPNet_Demo/utils/GTProcessing.py:177
Functioncv_imread
(filepath)
SLPNet_Demo/main.py:15
Functiondet_show
:param img_path: full image path for evaluation :param save_path: to save the eval image :param out_corner: corner after nms of output
module/det_part/PostProcessing.py:98
Functiondet_show
:param img_path: full image path for evaluation :param save_path: to save the eval image :param out_corner: corner after nms of output
SLPNet_Demo/utils/PostProcessing.py:98
Functiondetection_decoder
:param predict_maps: outputs from detection network, size(B, 32, 32, 12) :param top_k: :param value_threshold: :param nms_thresho
module/decoder.py:8
Functiondetection_decoder
:param predict_maps: outputs from detection network, size(B, 32, 32, 12) :param top_k: :param value_threshold: :param nms_thresho
SLPNet_Demo/utils/decoder.py:18
Functiondilate_3x3
(bool_grid_maps)
module/det_part/GTProcessing.py:195
Functiondilate_3x3
(bool_grid_maps)
SLPNet_Demo/utils/GTProcessing.py:195
Functiondistinguish_point_pos
:param corners_list: tensor(8), eight corner coordinate, clockwise :param point: tensor(N, 2), to be distinguished :return: bool tenso
module/det_part/GTProcessing.py:32
Functiondistinguish_point_pos
:param corners_list: tensor(8), eight corner coordinate, clockwise :param point: tensor(N, 2), to be distinguished :return: bool tenso
SLPNet_Demo/utils/GTProcessing.py:32
Methodforward
(self, x)
SSNet_framework.py:31
Methodforward
(self, x)
SSNet_framework.py:71
Methodforward
(self, det_maps, det_labels, log_probs, reg_labels, input_lengths, target_lengths, mode='det_reg')
module/total_loss.py:13
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
(self, output_maps, corners_list_512)
module/det_part/detection_head.py:272
Methodforward
(self, output_maps, corners_list_512)
module/det_part/detection_head.py:411
Methodforward
(self, output_maps, corners_list_512)
module/det_part/detection_head.py:507
Methodforward
(self, x)
model/basic_modules.py:73
Methodforward
(self, x)
model/basic_modules.py:94
Methodforward
(self, x)
model/basic_modules.py:148
Methodforward
(self, x)
model/basic_modules.py:214
Methodforward
(self, x)
model/basic_modules.py:259
Methodforward
(self, x_list)
model/basic_modules.py:292
Methodforward
(self, x)
model/basic_modules.py:332
Methodforward
(self, x)
model/basic_modules.py:379
Methodforward
(self, x, mode1='det_reg', mode2='eval')
model/detection_recognition_pipeline.py:28
Methodforward
(self, x)
model/SSNet_modules.py:68
Methodforward
(self, x)
model/SSNet_modules.py:155
Methodforward
(self, x)
model/SSNet_modules.py:228
Methodforward
(self, x, x_det1, x_det2)
model/SSNet_modules.py:309
Methodforward
(self, x, x_det)
model/SSNet_modules.py:390
Methodforward
(self, x)
model/other_models/shufflenet.py:116
Methodforward
(self, x_list)
model/other_models/shufflenet.py:151
Methodforward
(self, x)
model/other_models/shufflenet.py:223
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
Methodforward
(self, output_maps, corners_list_512)
SLPNet_Demo/utils/detection_head.py:248
Methodforward
(self, x)
SLPNet_Demo/model/basic_modules.py:73
Methodforward
(self, x)
SLPNet_Demo/model/basic_modules.py:94
Methodforward
(self, x)
SLPNet_Demo/model/basic_modules.py:148
Methodforward
(self, x)
SLPNet_Demo/model/basic_modules.py:214
Methodforward
(self, x)
SLPNet_Demo/model/basic_modules.py:259
Methodforward
(self, x_list)
SLPNet_Demo/model/basic_modules.py:292
Methodforward
(self, x)
SLPNet_Demo/model/basic_modules.py:332
Methodforward
(self, x)
SLPNet_Demo/model/basic_modules.py:379
Methodforward
(self, x, mode1='det_reg', mode2='eval')
SLPNet_Demo/model/detection_recognition_pipeline.py:18
Methodforward
(self, x)
SLPNet_Demo/model/SSNet_modules.py:69
Methodforward
(self, x)
SLPNet_Demo/model/SSNet_modules.py:156
Methodforward
(self, x)
SLPNet_Demo/model/SSNet_modules.py:229
Functiongaussian_eval
:param corner_preds: from function: detection_analysis, tensor size(obj1, 8) :param corner_targets: tensor, size(obj2, 8) :param gauss
module/det_part/PostProcessing.py:262
Functiongaussian_eval
:param corner_preds: from function: detection_analysis, tensor size(obj1, 8) :param corner_targets: tensor, size(obj2, 8) :param gauss
SLPNet_Demo/utils/PostProcessing.py:262
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
← previousnext →101–200 of 218, ranked by callers