MCPcopy Create free account

hub / github.com/Yibin122/TensorRT-LaneATT / functions

Functions62 in github.com/Yibin122/TensorRT-LaneATT

⨍ Functions 62◇ Types & classes 15
↓ 4 callersMethod__init__
(self, in_channels, out_channels, kernel_size,
ganet/deform_conv.py:210
↓ 1 callersMethodDetect
ganet/ganet.cc:42
↓ 1 callersMethodDetectLane
laneatt/laneatt.cc:39
↓ 1 callersFunctionLaneIoU
laneatt/laneatt.cc:86
↓ 1 callersMethodLogger
ganet/ganet.hh:9
↓ 1 callersMethod__str__
(self)
common.py:107
↓ 1 callersMethod_infer_shape
(ctx, input, weight)
ganet/deform_conv.py:192
↓ 1 callersMethod_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys
ganet/deform_conv.py:381
↓ 1 callersMethod_output_size
(input, weight, padding, dilation, stride)
ganet/deform_conv.py:99
↓ 1 callersFunctionbuild_engine
Takes an ONNX file and creates a TensorRT engine to run inference with
onnx_to_tensorrt.py:17
↓ 1 callersFunctiondo_nms
(proposals, conf_threshold=0.4, nms_thres=50., nms_topk=4)
onnx_to_tensorrt.py:56
↓ 1 callersFunctionengine_inference
Create a TensorRT engine for ONNX-based LaneATT and run inference.
onnx_to_tensorrt.py:112
↓ 1 callersFunctionexport_onnx
(onnx_file_path)
laneatt_to_onnx.py:65
↓ 1 callersFunctionexport_onnx
(onnx_file_path)
ganet/ganet_pth2onnx.py:39
↓ 1 callersFunctionget_data_path
(data_dir)
common.py:59
↓ 1 callersFunctionget_engine
Attempts to load a serialized engine if available, otherwise builds a new TensorRT engine and saves it.
onnx_to_tensorrt.py:15
↓ 1 callersMethodinit_offset
(self)
ganet/deform_conv.py:372
↓ 1 callersMethodinit_offset
(self)
ganet/deform_conv.py:492
↓ 1 callersFunctionlocate_files
Locates the specified files in the specified data directories. If a file exists in multiple data directories, the first directory is used.
common.py:74
↓ 1 callersFunctionpost_process
(img, proposals, n_offsets=72)
onnx_to_tensorrt.py:70
↓ 1 callersMethodreset_parameters
(self)
ganet/deform_conv.py:247
↓ 1 callersMethodreset_parameters
(self)
ganet/deform_conv.py:313
↓ 1 callersMethodreset_parameters
(self)
ganet/deform_conv.py:444
MethodGANet
ganet/ganet.cc:14
FunctionGiB
(val)
common.py:33
MethodLaneATT
laneatt/laneatt.cc:15
MethodLoadEngine
laneatt/laneatt.cc:61
MethodLoadEngine
ganet/ganet.cc:66
MethodLogger
laneatt/laneatt.hh:10
MethodPostProcess
laneatt/laneatt.cc:104
Method__init__
(self, model)
laneatt_to_onnx.py:7
Method__init__
(self, host_mem, device_mem)
common.py:103
Method__init__
(self, model)
ganet/ganet_pth2onnx.py:9
Method__init__
(self, in_channels, out_channels, kernel_size,
ganet/deform_conv.py:274
Method__init__
(self, *args, **kwargs)
ganet/deform_conv.py:358
Method__init__
(self, in_channels, out_channels, kernel_size,
ganet/deform_conv.py:411
Method__init__
(self, *args, **kwargs)
ganet/deform_conv.py:478
Method__repr__
(self)
common.py:110
Method_load_from_state_dict
(self, state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys
ganet/deform_conv.py:505
Functionadd_help
(description)
common.py:37
Functionallocate_buffers
(engine)
common.py:114
Methodbackward
(ctx, grad_output)
ganet/deform_conv.py:62
Methodbackward
(ctx, grad_output)
ganet/deform_conv.py:170
Functiondo_inference
(context, bindings, inputs, outputs, stream, batch_size=1)
common.py:136
Functiondo_inference_v2
(context, bindings, inputs, outputs, stream)
common.py:150
Functionfind_sample_data
Parses sample arguments. Args: description (str): Description of the sample. subfolder (str): The subfolder containing data r
common.py:42
Methodforward
(self, x)
laneatt_to_onnx.py:30
Methodforward
(self, img)
ganet/ganet_pth2onnx.py:15
Methodforward
(ctx, input, offset, weight, stride=1,
ganet/deform_conv.py:18
Methodforward
(ctx, input, offset, mask, weight,
ganet/deform_conv.py:134
Methodforward
(self, x, offset)
ganet/deform_conv.py:254
Methodforward
(self, x, offset)
ganet/deform_conv.py:320
Methodforward
(self, x)
ganet/deform_conv.py:376
Methodforward
(self, x, offset, mask)
ganet/deform_conv.py:453
Methodforward
(self, x)
ganet/deform_conv.py:496
Methodlog
laneatt/laneatt.hh:13
Methodlog
ganet/ganet.hh:12
Functionmain
laneatt/laneatt_test.cc:4
Functionmain
ganet/main.cc:4
Methodsymbolic
(g, input, offset, weight, stride, padding, dilation, groups, deformable_groups, im2col_step)
ganet/deform_conv.py:116
Method~GANet
ganet/ganet.cc:31
Method~LaneATT
laneatt/laneatt.cc:28