Index

A

Accuracy (class in hat.metrics.acc) AccuracyAttrMultiLabel (class in hat.metrics.acc) AccuracySeg (class in hat.metrics.acc) add_sin_difference() (hat.models.task_modules.fcos3d.loss.FCOS3DLoss static method) (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method) AddGaussianNoise (class in hat.data.transforms.keypoints) AddKeys (class in hat.data.transforms.common) AddTemporalFusion (class in hat.models.task_modules.view_fusion.temporal_fusion) AlbuImageOnlyTransform (class in hat.data.transforms.detection) Anchor3DGeneratorStride (class in hat.models.task_modules.lidar.anchor_generator) AnchorPostProcess (class in hat.models.base_modules.postprocess.anchor_postprocess)

annealing_cos() (hat.callbacks.lr_updater.OneCycleUpdater static method) (hat.callbacks.param_updater.CyclicOptimParamUpdater static method) ArgmaxPostprocess (class in hat.models.base_modules.postprocess.argmax_postprocess) Argoverse1Dataset (class in hat.data.datasets.argoverse_dataset) Argoverse1Packer (class in hat.data.datasets.argoverse_dataset) Argoverse1Sampler (class in hat.data.datasets.argoverse_dataset) ArgoverseMetric (class in hat.metrics.argoverse_metric) ArgoverseViz (class in hat.visualize.argoverse) assign_per_class() (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner method) assign_targets() (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner method) AssignSegLabel (class in hat.data.transforms.lidar_utils.lidar_transform_3d) AugmentHSV (class in hat.data.transforms.detection)

B

BaseModelProfiler (class in hat.profiler.model_profiler) BasicBatchProcessor (class in hat.engine.processors.processor) BatchTransformDataset (class in hat.data.datasets.batch_transform_dataset) BatchVoxelization (class in hat.models.task_modules.pointpillars.preprocess) BBoxSelector (class in hat.data.transforms.lidar) BBoxTargetGenerator (class in hat.models.base_modules.target.bbox_target) BevDeformableTemporalAttention (class in hat.models.base_modules.transformer_attentions) BevDetDecoder (class in hat.models.task_modules.view_fusion.decoder) BevEncoder (class in hat.models.task_modules.view_fusion.encoder) BevFeatureFlip (class in hat.data.transforms.multi_views) BevFeatureRotate (class in hat.data.transforms.multi_views) BevSegDecoder (class in hat.models.task_modules.view_fusion.decoder)

BevSpatialCrossAtten (class in hat.models.base_modules.transformer_attentions) BgrToYuv444 (class in hat.data.transforms.common) (class in horizon_plugin_pytorch.nn.bgr_to_yuv444) BgrToYuv444V2 (class in hat.data.transforms.common) BiFPN (class in hat.models.necks.bifpn) BMSegmentor (class in hat.models.structures.segmentor) box_coder (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner property) box_ndim (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner property) BoxJitter (class in hat.data.transforms.detection) build_aanet_volume() (hat.models.task_modules.stereonet.headplus.StereoNetHeadPlus method) build_concat_volume() (hat.models.task_modules.stereonet.head.StereoNetHead method) build_gwc_volume() (hat.models.task_modules.stereonet.head.StereoNetHead method) build_res_list() (hat.models.task_modules.detr3d.head.Detr3dHead method)

C

Calibration2QAT (class in hat.models.model_convert.converters) Calibrator (class in hat.engine.calibrator) CarfusionCroppedData (class in hat.data.datasets.carfusion_keypoints_dataset) CarfusionPackData (class in hat.data.datasets.carfusion_keypoints_dataset) CBGSDataset (class in hat.data.datasets.dataset_wrappers) centered_yuv2bgr() (in module horizon_plugin_pytorch.functional) centered_yuv2rgb() (in module horizon_plugin_pytorch.functional) CenterPointBBoxCoder (class in hat.models.task_modules.centerpoint.bbox_coders) CenterPointDecoder (class in hat.models.task_modules.centerpoint.decoder) CenterPointDetector (class in hat.models.structures.detectors.centerpoint) CenterPointHead (class in hat.models.task_modules.centerpoint.head) CenterPointLidarTarget (class in hat.models.task_modules.centerpoint.target) CenterPointLoss (class in hat.models.task_modules.centerpoint.loss) CenterPointPostProcess (class in hat.models.task_modules.centerpoint.post_process) CenterPointPreProcess (class in hat.models.task_modules.centerpoint.pre_process) CenterPointTarget (class in hat.models.task_modules.centerpoint.target) CEWithHardMining (class in hat.models.losses.cross_entropy_loss) CEWithLabelSmooth (class in hat.models.losses.cross_entropy_loss) CEWithWeightMap (class in hat.models.losses.cross_entropy_loss) CFTAuxHead (class in hat.models.task_modules.view_fusion.cft_transformer) CFTTransformer (class in hat.models.task_modules.view_fusion.cft_transformer) check_model() (in module horizon_plugin_pytorch.quantization) check_transform() (hat.data.transforms.detection.AlbuImageOnlyTransform method) CheckDeployDevice (class in hat.profiler.model_profiler) CheckFused (class in hat.profiler.model_profiler) Checkpoint (class in hat.callbacks.checkpoint) CheckShared (class in hat.profiler.model_profiler) ChunkShuffleDataset (class in hat.data.datasets.dataset_wrappers) Cityscapes (class in hat.data.datasets.cityscapes) class_name (hat.models.task_modules.lidar.anchor_generator.Anchor3DGeneratorStride property) Classifier (class in hat.models.structures.classifier) ClassWiseTrackIdEncoder (class in hat.models.base_modules.label_encoder) Clip (class in hat.data.transforms.gaze.gaze) ClsViz (class in hat.visualize.cls)

CocktailCollate (class in hat.data.collates.collates) CocktailVargNetV2 (class in hat.models.backbones.vargnetv2) Coco (class in hat.data.datasets.mscoco) COCODetectionMetric (class in hat.metrics.coco_detection) CocoFromImage (class in hat.data.datasets.mscoco) ColorJitter (class in hat.data.transforms.detection) CompareWeights (class in hat.profiler.model_profiler) compile_model() (in module horizon_plugin_pytorch.quantization) ComposeDataset (class in hat.data.datasets.dataset_wrappers) compute() (hat.metrics.argoverse_metric.ArgoverseMetric method) (hat.metrics.mean_iou.MeanIOU method) (hat.metrics.metric_lane_detection.CulaneF1Score method) (hat.metrics.nuscenes_metric.NuscenesMetric method) (hat.metrics.voc_detection.VOCMApMetric method) ConcatDataset (class in hat.data.datasets.dataset_wrappers) Contrast (class in hat.data.transforms.faceid) convert() (in module horizon_plugin_pytorch.quantization.quantize) convert_fx() (in module horizon_plugin_pytorch.quantization.quantize_fx) ConvertDataType (class in hat.data.transforms.common) ConvertLayout (class in hat.data.transforms.classification) CoordConv (class in hat.models.task_modules.seg.utils) CopyKeys (class in hat.data.transforms.common) CosineAnnealingLrUpdater (class in hat.callbacks.lr_updater) CosLrUpdater (class in hat.callbacks.lr_updater) CosOptimParamUpdater (class in hat.callbacks.param_updater) CPUMemoryProfiler (class in hat.profiler.memory_profiler) create_targets_single() (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner method) CrossEntropyLoss (class in hat.models.losses.cross_entropy_loss) CuLaneDataset (class in hat.data.datasets.culane_dataset) CulaneF1Score (class in hat.metrics.metric_lane_detection) CuLaneFromImage (class in hat.data.datasets.culane_dataset) CuLanePacker (class in hat.data.datasets.culane_dataset) CyclicLrUpdater (class in hat.callbacks.lr_updater) CyclicOptimParamUpdater (class in hat.callbacks.param_updater)

D

DataParallelTrainer (class in hat.engine.dp_trainer) DBFilterByDifficulty (class in hat.data.transforms.lidar_utils.preprocess) DBFilterByMinNumPoint (class in hat.data.transforms.lidar_utils.preprocess) decode() (hat.models.task_modules.centerpoint.bbox_coders.CenterPointBBoxCoder method) (hat.models.task_modules.fcos3d.bbox_coder.FCOS3DBBoxCoder method) (hat.models.task_modules.lidar.box_coders.GroundBox3dCoder method) decode_yaw() (hat.models.task_modules.fcos3d.bbox_coder.FCOS3DBBoxCoder static method) DeconvDecoder (class in hat.models.task_modules.carfusion_keypoints.keypoint_head) Deeplabv3plusHead (class in hat.models.task_modules.deeplab.head) DeleteKeys (class in hat.data.transforms.common) Densetnt (class in hat.models.task_modules.motion_forecasting.decoders.densetnt.head) DensetntLoss (class in hat.models.task_modules.motion_forecasting.decoders.densetnt.loss) DensetntPostprocess (class in hat.models.task_modules.motion_forecasting.decoders.densetnt.post_process) DensetntTarget (class in hat.models.task_modules.motion_forecasting.decoders.densetnt.target) DepthwiseSeparableCenterPointHead (class in hat.models.task_modules.centerpoint.head) DepthwiseSeparableFCNHead (class in hat.models.task_modules.fcn.head) DequantModule (class in hat.models.base_modules.dequant_module) describe() (hat.profiler.memory_profiler.StageCPUMemoryProfiler method) DetAffineAugTransformer (class in hat.data.transforms.detection) DetectionPostProcess (class in horizon_plugin_pytorch.nn.detection_post_process) DetectionPostProcessV1 (class in horizon_plugin_pytorch.nn.detection_post_process_v1) DetectionTargetGenerator (class in hat.data.transforms.lidar) DetMosaic (class in hat.data.transforms.detection) Detr (class in hat.models.structures.detectors.detr)

Detr3d (class in hat.models.structures.detectors.detr3d) Detr3dDecoder (class in hat.models.task_modules.detr3d.head) Detr3dHead (class in hat.models.task_modules.detr3d.head) Detr3dPostProcess (class in hat.models.task_modules.detr3d.post_process) Detr3dTarget (class in hat.models.task_modules.detr3d.target) Detr3dTransformer (class in hat.models.task_modules.detr3d.head) DetrCriterion (class in hat.models.task_modules.detr.criterion) DetrHead (class in hat.models.task_modules.detr.head) DetrPostProcess (class in hat.models.task_modules.detr.post_process) DetViz (class in hat.visualize.det) DetYOLOv5MixUp (class in hat.data.transforms.detection) dis_mul() (hat.models.task_modules.stereonet.head.StereoNetHead method) (hat.models.task_modules.stereonet.headplus.StereoNetHeadPlus method) dis_sum() (hat.models.task_modules.stereonet.head.StereoNetHead method) (hat.models.task_modules.stereonet.headplus.StereoNetHeadPlus method) DispViz (class in hat.visualize.disparity) DistributedComposeRandomDataset (class in hat.data.datasets.dataset_wrappers) DistributedCycleMultiDatasetSampler (class in hat.data.samplers.dist_cycle_sampler_multi_dataset) DistributedDataParallelTrainer (class in hat.engine.ddp_trainer) DistributedGroupSampler (class in hat.data.samplers.dist_group_sampler) DistSamplerHook (class in hat.data.samplers.dist_sampler) DistSetEpochDatasetSampler (class in hat.data.samplers.dist_set_epoch_dataset_sampler) DumpData (class in hat.callbacks.dump_data) DwUnet (class in hat.models.necks.dw_unet) DynamicFcosTarget (class in hat.models.task_modules.fcos.target)

E

EfficientNet (class in hat.models.backbones.efficientnet) efficientnet() (in module hat.models.backbones.efficientnet) efficientnet_lite() (in module hat.models.backbones.efficientnet) ElementwiseL1HingeLoss (class in hat.models.losses.hinge_loss) ElementwiseL2HingeLoss (class in hat.models.losses.hinge_loss) encode() (hat.models.task_modules.lidar.box_coders.GroundBox3dCoder method) EncoderDecoder (class in hat.models.structures.encoder_decoder) EndPointError (class in hat.metrics.metric_optical_flow) ExponentialMovingAverage (class in hat.callbacks.exponential_moving_average) export_hbir() (in module horizon_plugin_pytorch.quantization) export_reference_points() (hat.models.structures.detectors.detr3d.Detr3d method) (hat.models.structures.view_fusion.ViewFusion method) (hat.models.task_modules.petr.head.PETRHead method) (hat.models.task_modules.view_fusion.cft_transformer.CFTTransformer method)

ExtendVarGNetFeatures (class in hat.models.base_modules.basic_vargnet_module) extra_repr() (horizon_plugin_pytorch.quantization.fake_quantize.FakeQuantize method) extract_feat() (hat.models.structures.detectors.detr.Detr method) (hat.models.structures.detectors.detr3d.Detr3d method) (hat.models.structures.detectors.fcos.FCOS method) (hat.models.structures.detectors.fcos3d.FCOS3D method) (hat.models.structures.track_pred.motr.Motr method) ExtSequential (class in hat.models.base_modules.extend_container)

F

FakeQuantize (class in horizon_plugin_pytorch.quantization.fake_quantize) FastSCNNNeck (class in hat.models.necks.fast_scnn) FCNDecoder (class in hat.models.task_modules.fcn.decoder) FCNHead (class in hat.models.task_modules.fcn.head) FCNTarget (class in hat.models.task_modules.fcn.target) FCOS (class in hat.models.structures.detectors.fcos) FCOS3D (class in hat.models.structures.detectors.fcos3d) FCOS3DBBoxCoder (class in hat.models.task_modules.fcos3d.bbox_coder) FCOS3DLoss (class in hat.models.task_modules.fcos3d.loss) FCOS3DPostProcess (class in hat.models.task_modules.fcos3d.post_process) FCOS3DTarget (class in hat.models.task_modules.fcos3d.target) FCOSDecoder (class in hat.models.task_modules.fcos.decoder) FCOSDecoder4RCNN (class in hat.models.task_modules.fcos.decoder) FCOSHead (class in hat.models.task_modules.fcos.head) FCOSLoss (class in hat.models.task_modules.fcos.fcos_loss) FCOSMultiStrideCatFilter (class in hat.models.task_modules.fcos.filter) FCOSMultiStrideFilter (class in hat.models.task_modules.fcos.filter) FCOSTarget (class in hat.models.task_modules.fcos.target) FCOSTarget4RPNHead (class in hat.models.task_modules.fcos.target) FeaturemapSimilarity (class in hat.profiler.model_profiler) filter() (in module horizon_plugin_pytorch.nn.functional) fit() (hat.engine.predictor.Predictor method) FixedCrop (class in hat.data.transforms.detection) FixWeightQScale (class in hat.models.model_convert.converters) Float2Calibration (class in hat.models.model_convert.converters) Float2QAT (class in hat.models.model_convert.converters) FloatQatConvertPipeline (class in hat.models.model_convert.pipelines) FlowViz (class in hat.visualize.opticalflow) FlyingChairs (class in hat.data.datasets.flyingchairs_dataset) FlyingChairsFromImage (class in hat.data.datasets.flyingchairs_dataset) FlyingChairsPacker (class in hat.data.datasets.flyingchairs_dataset) FocalLoss (class in hat.models.losses.focal_loss) FocalLossV2 (class in hat.models.losses.focal_loss) forward() (hat.data.transforms.segmentation.SegResize method) (hat.models.backbones.efficientnet.EfficientNet method) (hat.models.backbones.horizon_swin_transformer.HorizonSwinTransformer method) (hat.models.backbones.mixvargenet.MixVarGENet method) (hat.models.backbones.mobilenetv1.MobileNetV1 method) (hat.models.backbones.mobilenetv2.MobileNetV2 method) (hat.models.backbones.resnet.ResNetV2 method) (hat.models.backbones.vargconvnet.VargConvNet method) (hat.models.backbones.vargnetv2.CocktailVargNetV2 method) (hat.models.backbones.vargnetv2.VargNetV2 method) (hat.models.base_modules.basic_vargnet_module.ExtendVarGNetFeatures method) (hat.models.base_modules.bbox_decoder.XYWHBBoxDecoder method) (hat.models.base_modules.dequant_module.DequantModule method) (hat.models.base_modules.label_encoder.ClassWiseTrackIdEncoder method) (hat.models.base_modules.label_encoder.MatchLabelGroundLineEncoder method) (hat.models.base_modules.label_encoder.MatchLabelSepEncoder method) (hat.models.base_modules.label_encoder.MatchLabelTrackEncoder method) (hat.models.base_modules.label_encoder.MultiClassMatchLabelSepEncoder method) (hat.models.base_modules.label_encoder.OneHotClassEncoder method) (hat.models.base_modules.label_encoder.PersonPositionLabelFromMatch method) (hat.models.base_modules.label_encoder.RCNN3DLabelFromMatch method) (hat.models.base_modules.label_encoder.RCNNBinDetLabelFromMatch method) (hat.models.base_modules.label_encoder.RCNNKPSLabelFromMatch method) (hat.models.base_modules.label_encoder.XYWHBBoxEncoder method) (hat.models.base_modules.matcher.IgRegionMatcher method) (hat.models.base_modules.matcher.MaxIoUMatcher method) (hat.models.base_modules.position_encoding.LearnedPositionalEncoding method) (hat.models.base_modules.postprocess.anchor_postprocess.AnchorPostProcess method) (hat.models.base_modules.postprocess.argmax_postprocess.ArgmaxPostprocess method) (hat.models.base_modules.postprocess.argmax_postprocess.HorizonAdasClsPostProcessor method) (hat.models.base_modules.postprocess.max_postprocess.MaxPostProcess method) (hat.models.base_modules.postprocess.rle_postprocess.RLEPostprocess method) (hat.models.base_modules.quant_module.QuantModule method) (hat.models.base_modules.resize_parser.ResizeParser method) (hat.models.base_modules.roi_feat_extractors.MultiScaleRoIAlign method) (hat.models.base_modules.target.bbox_target.BBoxTargetGenerator method) (hat.models.base_modules.target.bbox_target.ProposalTargetGroundLine method) (hat.models.base_modules.target.bbox_target.ProposalTargetTrack method) (hat.models.base_modules.target.heatmap_roi_3d_target.HeatMap3DTargetGenerator method) (hat.models.base_modules.transformer_attentions.BevDeformableTemporalAttention method) (hat.models.base_modules.transformer_attentions.BevSpatialCrossAtten method) (hat.models.base_modules.transformer_attentions.MSDeformableAttention3D method) (hat.models.base_modules.transformer_attentions.ObjectDetr3DCrossAtten method) (hat.models.embeddings.PositionEmbeddingLearned method) (hat.models.embeddings.PositionEmbeddingSine method) (hat.models.losses.cross_entropy_loss.CEWithHardMining method) (hat.models.losses.cross_entropy_loss.CEWithLabelSmooth method) (hat.models.losses.cross_entropy_loss.CEWithWeightMap method) (hat.models.losses.cross_entropy_loss.CrossEntropyLoss method) (hat.models.losses.cross_entropy_loss.SoftTargetCrossEntropy method) (hat.models.losses.focal_loss.FocalLoss method) (hat.models.losses.focal_loss.FocalLossV2 method) (hat.models.losses.focal_loss.GaussianFocalLoss method) (hat.models.losses.focal_loss.SoftmaxFocalLoss method) (hat.models.losses.giou_loss.GIoULoss method) (hat.models.losses.hinge_loss.WeightedSquaredHingeLoss method) (hat.models.losses.l1_loss.L1Loss method) (hat.models.losses.lnnorm_loss.LnNormLoss method) (hat.models.losses.mse_loss.MSELoss method) (hat.models.losses.seg_loss.MixSegLoss method) (hat.models.losses.seg_loss.MixSegLossMultipreds method) (hat.models.losses.seg_loss.MultiStrideLosses method) (hat.models.losses.seg_loss.SegEdgeLoss method) (hat.models.losses.seg_loss.SegLoss method) (hat.models.losses.smooth_l1_loss.SmoothL1Loss method) (hat.models.losses.yolo_losses.YOLOV3Loss method) (hat.models.necks.bifpn.BiFPN method) (hat.models.necks.dw_unet.DwUnet method) (hat.models.necks.fast_scnn.FastSCNNNeck method) (hat.models.necks.fpn.FPN method) (hat.models.necks.pafpn.PAFPN method) (hat.models.necks.pafpn.VargPAFPN method) (hat.models.necks.retinanet_fpn.RetinaNetFPN method) (hat.models.necks.second_neck.SECONDNeck method) (hat.models.necks.unet.Unet method) (hat.models.necks.yolov3.YOLOV3Neck method) (hat.models.necks.yolov3_group.YoloGroupNeck method) (hat.models.structures.classifier.Classifier method) (hat.models.structures.detectors.centerpoint.CenterPointDetector method) (hat.models.structures.detectors.detr.Detr method) (hat.models.structures.detectors.detr3d.Detr3d method) (hat.models.structures.detectors.fcos.FCOS method) (hat.models.structures.detectors.fcos3d.FCOS3D method) (hat.models.structures.detectors.pointpillars.PointPillarsDetector method) (hat.models.structures.detectors.retinanet.RetinaNet method) (hat.models.structures.detectors.yolov3.YOLOV3 method) (hat.models.structures.disparity_pred.stereonet.StereoNet method) (hat.models.structures.disparity_pred.stereonet.StereoNetPlus method) (hat.models.structures.encoder_decoder.EncoderDecoder method) (hat.models.structures.keypoints.keypoint_model.HeatmapKeypointModel method) (hat.models.structures.lane_pred.ganet.GaNet method) (hat.models.structures.lidar_multitask.lidar_multitask.LidarMultiTask method) (hat.models.structures.motion_forecasting.MotionForecasting method) (hat.models.structures.opticalflow.pwcnet.PwcNet method) (hat.models.structures.segmentor.BMSegmentor method) (hat.models.structures.segmentor.Segmentor method) (hat.models.structures.segmentor.SegmentorV2 method) (hat.models.structures.track_pred.motr.Motr method) (hat.models.structures.view_fusion.ViewFusion method) (hat.models.task_modules.carfusion_keypoints.heatmap_decoder.HeatmapDecoder method) (hat.models.task_modules.carfusion_keypoints.keypoint_head.DeconvDecoder method) (hat.models.task_modules.centerpoint.decoder.CenterPointDecoder method) (hat.models.task_modules.centerpoint.head.CenterPointHead method) (hat.models.task_modules.centerpoint.loss.CenterPointLoss method) (hat.models.task_modules.centerpoint.post_process.CenterPointPostProcess method) (hat.models.task_modules.centerpoint.pre_process.CenterPointPreProcess method) (hat.models.task_modules.centerpoint.target.CenterPointLidarTarget method) (hat.models.task_modules.centerpoint.target.CenterPointTarget method) (hat.models.task_modules.deeplab.head.Deeplabv3plusHead method) (hat.models.task_modules.detr.criterion.DetrCriterion method) (hat.models.task_modules.detr.head.DetrHead method) (hat.models.task_modules.detr.post_process.DetrPostProcess method) (hat.models.task_modules.detr.transformer.Transformer method) (hat.models.task_modules.detr3d.head.Detr3dDecoder method) (hat.models.task_modules.detr3d.head.Detr3dHead method) (hat.models.task_modules.detr3d.head.Detr3dTransformer method) (hat.models.task_modules.detr3d.post_process.Detr3dPostProcess method) (hat.models.task_modules.detr3d.target.Detr3dTarget method) (hat.models.task_modules.fcn.decoder.FCNDecoder method) (hat.models.task_modules.fcn.head.FCNHead method) (hat.models.task_modules.fcn.target.FCNTarget method) (hat.models.task_modules.fcos.decoder.FCOSDecoder method) (hat.models.task_modules.fcos.decoder.FCOSDecoder4RCNN method) (hat.models.task_modules.fcos.decoder.VehicleSideFCOSDecoder method) (hat.models.task_modules.fcos.fcos_loss.FCOSLoss method) (hat.models.task_modules.fcos.fcos_loss.VehicleSideFCOSLoss method) (hat.models.task_modules.fcos.filter.FCOSMultiStrideCatFilter method) (hat.models.task_modules.fcos.filter.FCOSMultiStrideFilter method) (hat.models.task_modules.fcos.head.FCOSHead method) (hat.models.task_modules.fcos.head.VehicleSideFCOSHead method) (hat.models.task_modules.fcos.target.DynamicFcosTarget method) (hat.models.task_modules.fcos.target.FCOSTarget method) (hat.models.task_modules.fcos.target.VehicleSideFCOSTarget method) (hat.models.task_modules.fcos3d.loss.FCOS3DLoss method) (hat.models.task_modules.fcos3d.post_process.FCOS3DPostProcess method) (hat.models.task_modules.fcos3d.target.FCOS3DTarget method) (hat.models.task_modules.ganet.decoder.GaNetDecoder method) (hat.models.task_modules.ganet.head.GaNetHead method) (hat.models.task_modules.ganet.losses.GaNetLoss method) (hat.models.task_modules.ganet.neck.GaNetNeck method) (hat.models.task_modules.ganet.target.GaNetTarget method) (hat.models.task_modules.lidar.anchor_generator.Anchor3DGeneratorStride method) (hat.models.task_modules.lidar.pillar_encoder.PillarFeatureNet method) (hat.models.task_modules.lidar.pillar_encoder.PointPillarScatter method) (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner method) (hat.models.task_modules.lidar_multitask.decoder.LidarSegDecoder method) (hat.models.task_modules.motion_forecasting.decoders.densetnt.head.Densetnt method) (hat.models.task_modules.motion_forecasting.decoders.densetnt.loss.DensetntLoss method) (hat.models.task_modules.motion_forecasting.decoders.densetnt.post_process.DensetntPostprocess method) (hat.models.task_modules.motion_forecasting.decoders.densetnt.target.DensetntTarget method) (hat.models.task_modules.motion_forecasting.encoders.vectornet.Vectornet method) (hat.models.task_modules.motr.criterion.MotrCriterion method) (hat.models.task_modules.motr.head.MotrHead method) (hat.models.task_modules.motr.motr_deformable_transformer.MotrDeformableTransformer method) (hat.models.task_modules.motr.post_process.MotrPostProcess method) (hat.models.task_modules.motr.qim.QueryInteractionModule method) (hat.models.task_modules.petr.head.PETRDecoder method) (hat.models.task_modules.petr.head.PETRHead method) (hat.models.task_modules.petr.head.PETRTransformer method) (hat.models.task_modules.pointpillars.head.PointPillarsHead method) (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method) (hat.models.task_modules.pointpillars.postprocess.PointPillarsPostProcess method) (hat.models.task_modules.pointpillars.preprocess.BatchVoxelization method) (hat.models.task_modules.pointpillars.preprocess.PointPillarsPreProcess method) (hat.models.task_modules.pwcnet.head.PwcNetHead method) (hat.models.task_modules.pwcnet.neck.PwcNetNeck method) (hat.models.task_modules.retinanet.filter.RetinanetMultiStrideFilter method) (hat.models.task_modules.retinanet.head.RetinaNetHead method) (hat.models.task_modules.retinanet.postprocess.RetinaNetPostProcess method) (hat.models.task_modules.seg.decoder.VargNetSegDecoder method) (hat.models.task_modules.seg.head.SegHead method) (hat.models.task_modules.seg.utils.CoordConv method) (hat.models.task_modules.seg.vargnet_seg_head.FRCNNSegHead method) (hat.models.task_modules.stereonet.head.StereoNetHead method) (hat.models.task_modules.stereonet.headplus.StereoNetHeadPlus method) (hat.models.task_modules.stereonet.neck.StereoNetNeck method) (hat.models.task_modules.stereonet.post_process.StereoNetPostProcess method) (hat.models.task_modules.stereonet.post_process.StereoNetPostProcessPlus method) (hat.models.task_modules.view_fusion.cft_transformer.CFTAuxHead method) (hat.models.task_modules.view_fusion.cft_transformer.CFTTransformer method) (hat.models.task_modules.view_fusion.encoder.BevEncoder method) (hat.models.task_modules.yolo.anchor.YOLOV3AnchorGenerator method) (hat.models.task_modules.yolo.filter.YOLOv3Filter method) (hat.models.task_modules.yolo.head.YOLOV3Head method) (hat.models.task_modules.yolo.label_encoder.YOLOV3LabelEncoder method) (hat.models.task_modules.yolo.matcher.YOLOV3Matcher method) (hat.models.task_modules.yolo.postprocess.YOLOV3PostProcess method) (horizon_plugin_pytorch.nn.bgr_to_yuv444.BgrToYuv444 method) (horizon_plugin_pytorch.nn.detection_post_process.DetectionPostProcess method) (horizon_plugin_pytorch.nn.detection_post_process_v1.DetectionPostProcessV1 method) (horizon_plugin_pytorch.quantization.fake_quantize.FakeQuantize method) (horizon_plugin_pytorch.quantization.observer.MovingAverageMinMaxObserver method) (horizon_plugin_pytorch.quantization.observer.MovingAveragePerChannelMinMaxObserver method)

forward_single() (hat.models.task_modules.detr.head.DetrHead method) (hat.models.task_modules.fcos.head.FCOSHead method) (hat.models.task_modules.fcos.head.VehicleSideFCOSHead method) (hat.models.task_modules.retinanet.head.RetinaNetHead method) (hat.models.task_modules.seg.head.SegHead method) FPN (class in hat.models.necks.fpn) FRCNNSegHead (class in hat.models.task_modules.seg.vargnet_seg_head) FreezeBNStatistics (class in hat.callbacks.freeze_bn) FreezeModule (class in hat.callbacks.online_model_trick) fuse_fx() (in module horizon_plugin_pytorch.quantization.quantize_fx) fuse_known_modules() (in module horizon_plugin_pytorch.quantization.fuse_modules) fuse_model() (hat.models.structures.detectors.centerpoint.CenterPointDetector method) (hat.models.structures.detectors.detr3d.Detr3d method) (hat.models.structures.detectors.pointpillars.PointPillarsDetector method) (hat.models.structures.disparity_pred.stereonet.StereoNet method) (hat.models.structures.lidar_multitask.lidar_multitask.LidarMultiTask method) (hat.models.structures.view_fusion.ViewFusion method) (hat.models.task_modules.centerpoint.head.CenterPointHead method) (hat.models.task_modules.detr3d.head.Detr3dDecoder method) (hat.models.task_modules.detr3d.head.Detr3dHead method) (hat.models.task_modules.detr3d.head.Detr3dTransformer method) (hat.models.task_modules.petr.head.PETRDecoder method) (hat.models.task_modules.petr.head.PETRHead method) (hat.models.task_modules.petr.head.PETRTransformer method) (hat.models.task_modules.stereonet.head.StereoNetHead method) (hat.models.task_modules.stereonet.neck.StereoNetNeck method) (hat.models.task_modules.view_fusion.encoder.BevEncoder method) (hat.models.task_modules.view_fusion.view_transformer.LSSTransformer method) fuse_modules() (in module horizon_plugin_pytorch.quantization.fuse_modules) FuseBN (class in hat.callbacks.online_model_trick)

G

GaNet (class in hat.models.structures.lane_pred.ganet) GaNetDecoder (class in hat.models.task_modules.ganet.decoder) GaNetHead (class in hat.models.task_modules.ganet.head) GaNetLoss (class in hat.models.task_modules.ganet.losses) GaNetNeck (class in hat.models.task_modules.ganet.neck) GaNetTarget (class in hat.models.task_modules.ganet.target) gather_metrics() (hat.metrics.voc_detection.VOCMApMetric method) GaussianBlur (class in hat.data.transforms.faceid) GaussianFocalLoss (class in hat.models.losses.focal_loss) GazeRandomCropWoResize (class in hat.data.transforms.gaze.gaze) GazeRotate3DWithCrop (class in hat.data.transforms.gaze.gaze) GazeYUVTransform (class in hat.data.transforms.gaze.gaze) gen_reference_point() (hat.models.task_modules.view_fusion.view_transformer.LSSTransformer method) generate_anchors() (hat.models.task_modules.lidar.anchor_generator.Anchor3DGeneratorStride method) GenerateHeatmapTarget (class in hat.data.transforms.keypoints) get() (hat.metrics.argoverse_metric.ArgoverseMetric method) (hat.metrics.coco_detection.COCODetectionMetric method) (hat.metrics.kitti2d_detection.Kitti2DMetric method) (hat.metrics.kitti3d_detection.Kitti3DMetricDet method) (hat.metrics.loss_show.LossShow method) (hat.metrics.mot_metrics.MotMetric method) (hat.metrics.nuscenes_metric.NuscenesMetric method) (hat.metrics.nuscenes_metric.NuscenesMonoMetric method) get_ade() (hat.metrics.argoverse_metric.ArgoverseMetric method) get_attr_name() (hat.metrics.nuscenes_metric.NuscenesMonoMetric method) get_box_reg_loss() (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method) get_cat_ids() (hat.data.datasets.nuscenes_dataset.NuscenesLidarDataset method) get_cls_loss() (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method) get_crop_bbox() (hat.data.transforms.segmentation.SegRandomCrop method) get_direction_target() (hat.models.task_modules.fcos3d.loss.FCOS3DLoss static method) (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method)

get_fde() (hat.metrics.argoverse_metric.ArgoverseMetric method) get_indexes() (hat.data.transforms.detection.DetMosaic method) (hat.data.transforms.detection.DetYOLOv5MixUp method) get_intersections_to_vertical() (hat.models.base_modules.label_encoder.MatchLabelGroundLineEncoder static method) get_l_img() (hat.models.task_modules.stereonet.headplus.StereoNetHeadPlus method) get_label() (hat.models.base_modules.label_encoder.RCNNBinDetLabelFromMatch method) get_lr() (hat.callbacks.lr_updater.CosineAnnealingLrUpdater method) (hat.callbacks.lr_updater.CosLrUpdater method) (hat.callbacks.lr_updater.CyclicLrUpdater method) (hat.callbacks.lr_updater.NoamLrUpdater method) (hat.callbacks.lr_updater.OneCycleUpdater method) (hat.callbacks.lr_updater.PolyLrUpdater method) (hat.callbacks.lr_updater.StepDecayLrUpdater method) get_offset() (hat.models.task_modules.stereonet.headplus.StereoNetHeadPlus method) get_param() (hat.callbacks.param_updater.CosOptimParamUpdater method) (hat.callbacks.param_updater.CyclicOptimParamUpdater method) (hat.callbacks.param_updater.StepDecayOptimParamUpdater method) get_pos_neg_loss() (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method) get_score_map() (hat.models.base_modules.label_encoder.RCNNKPSLabelFromMatch method) get_targets_single() (hat.models.task_modules.centerpoint.target.CenterPointLidarTarget method) (hat.models.task_modules.view_fusion.cft_transformer.CFTAuxHead method) get_task_detections() (hat.models.task_modules.centerpoint.post_process.CenterPointPostProcess method) get_warmup_lr() (hat.callbacks.lr_updater.NoamLrUpdater method) (hat.callbacks.lr_updater.OneCycleUpdater method) GIoULoss (class in hat.models.losses.giou_loss) GKTTransformer (class in hat.models.task_modules.view_fusion.view_transformer) GPUMemoryProfiler (class in hat.profiler.memory_profiler) GradClip (class in hat.callbacks.grad_scale) GradScale (class in hat.callbacks.grad_scale) GridMask (class in hat.data.transforms.grid_mask) GroundBox3dCoder (class in hat.models.task_modules.lidar.box_coders) groupwise_correlation() (hat.models.task_modules.stereonet.head.StereoNetHead method)

H

hat.callbacks.checkpoint module hat.callbacks.dump_data module hat.callbacks.exponential_moving_average module hat.callbacks.freeze_bn module hat.callbacks.grad_scale module hat.callbacks.lr_updater module hat.callbacks.metric_updater module hat.callbacks.monitor module hat.callbacks.online_model_trick module hat.callbacks.param_updater module hat.callbacks.save_traced module hat.callbacks.tensorboard module hat.callbacks.validation module hat.data.collates.collates module hat.data.dataloaders.passthrough_dataloader module hat.data.datasets.argoverse_dataset module hat.data.datasets.batch_transform_dataset module hat.data.datasets.carfusion_keypoints_dataset module hat.data.datasets.cityscapes module hat.data.datasets.culane_dataset module hat.data.datasets.dataset_wrappers module hat.data.datasets.flyingchairs_dataset module hat.data.datasets.imagenet module hat.data.datasets.kitti2d module hat.data.datasets.kitti3d module hat.data.datasets.mot17_dataset module hat.data.datasets.mscoco module hat.data.datasets.nuscenes_dataset module hat.data.datasets.rand_dataset module hat.data.datasets.sceneflow_dataset module hat.data.datasets.voc module hat.data.samplers.dist_cycle_sampler_multi_dataset module hat.data.samplers.dist_group_sampler module hat.data.samplers.dist_sampler module hat.data.samplers.dist_set_epoch_dataset_sampler module hat.data.samplers.selected_sampler module hat.data.transforms.classification module hat.data.transforms.common module hat.data.transforms.detection module hat.data.transforms.faceid module hat.data.transforms.gaze.gaze module hat.data.transforms.grid_mask module hat.data.transforms.keypoints module hat.data.transforms.lidar module hat.data.transforms.lidar_utils.lidar_transform_3d module hat.data.transforms.lidar_utils.preprocess module hat.data.transforms.lidar_utils.sample_ops module hat.data.transforms.multi_views module hat.data.transforms.segmentation module hat.data.transforms.seq_transform module hat.engine.calibrator module hat.engine.ddp_trainer module hat.engine.dp_trainer module hat.engine.predictor module hat.engine.processors.processor module hat.engine.trainer module hat.metrics.acc module hat.metrics.argoverse_metric module hat.metrics.coco_detection module hat.metrics.kitti2d_detection module hat.metrics.kitti3d_detection module hat.metrics.loss_show module hat.metrics.mean_iou module hat.metrics.metric_keypoints module hat.metrics.metric_lane_detection module hat.metrics.metric_optical_flow module hat.metrics.mot_metrics module hat.metrics.nuscenes_metric module hat.metrics.voc_detection module hat.models.backbones.contrib.resnet module hat.models.backbones.efficientnet module hat.models.backbones.horizon_swin_transformer module hat.models.backbones.mixvargenet module hat.models.backbones.mobilenetv1 module hat.models.backbones.mobilenetv2 module hat.models.backbones.resnet module hat.models.backbones.vargconvnet module hat.models.backbones.vargdarknet module hat.models.backbones.vargnetv2 module hat.models.base_modules.basic_vargnet_module module hat.models.base_modules.bbox_decoder module hat.models.base_modules.dequant_module module hat.models.base_modules.extend_container module hat.models.base_modules.label_encoder module hat.models.base_modules.matcher module hat.models.base_modules.position_encoding module hat.models.base_modules.postprocess.anchor_postprocess module hat.models.base_modules.postprocess.argmax_postprocess module hat.models.base_modules.postprocess.max_postprocess module hat.models.base_modules.postprocess.rle_postprocess module hat.models.base_modules.quant_module module hat.models.base_modules.resize_parser module hat.models.base_modules.roi_feat_extractors module hat.models.base_modules.target.bbox_target module hat.models.base_modules.target.heatmap_roi_3d_target module hat.models.base_modules.target.reshape_target module hat.models.base_modules.transformer_attentions module hat.models.base_modules.transformer_bricks module hat.models.embeddings module hat.models.losses.cross_entropy_loss module hat.models.losses.focal_loss module hat.models.losses.giou_loss module hat.models.losses.hinge_loss module hat.models.losses.l1_loss module hat.models.losses.lnnorm_loss module hat.models.losses.mse_loss module hat.models.losses.seg_loss module hat.models.losses.smooth_l1_loss module hat.models.losses.yolo_losses module hat.models.model_convert.converters module hat.models.model_convert.pipelines module hat.models.necks.bifpn module hat.models.necks.dw_unet module hat.models.necks.fast_scnn module hat.models.necks.fpn module hat.models.necks.pafpn module hat.models.necks.retinanet_fpn module hat.models.necks.second_neck module hat.models.necks.unet module hat.models.necks.yolov3 module hat.models.necks.yolov3_group module hat.models.structures.classifier module hat.models.structures.detectors.centerpoint module hat.models.structures.detectors.detr module hat.models.structures.detectors.detr3d module

hat.models.structures.detectors.fcos module hat.models.structures.detectors.fcos3d module hat.models.structures.detectors.pointpillars module hat.models.structures.detectors.retinanet module hat.models.structures.detectors.yolov3 module hat.models.structures.disparity_pred.stereonet module hat.models.structures.encoder_decoder module hat.models.structures.keypoints.keypoint_model module hat.models.structures.lane_pred.ganet module hat.models.structures.lidar_multitask.lidar_multitask module hat.models.structures.motion_forecasting module hat.models.structures.opticalflow.pwcnet module hat.models.structures.segmentor module hat.models.structures.track_pred.motr module hat.models.structures.view_fusion module hat.models.task_modules.carfusion_keypoints.heatmap_decoder module hat.models.task_modules.carfusion_keypoints.keypoint_head module hat.models.task_modules.centerpoint.bbox_coders module hat.models.task_modules.centerpoint.decoder module hat.models.task_modules.centerpoint.head module hat.models.task_modules.centerpoint.loss module hat.models.task_modules.centerpoint.post_process module hat.models.task_modules.centerpoint.pre_process module hat.models.task_modules.centerpoint.target module hat.models.task_modules.deeplab.head module hat.models.task_modules.detr.criterion module hat.models.task_modules.detr.head module hat.models.task_modules.detr.post_process module hat.models.task_modules.detr.transformer module hat.models.task_modules.detr3d.head module hat.models.task_modules.detr3d.post_process module hat.models.task_modules.detr3d.target module hat.models.task_modules.fcn.decoder module hat.models.task_modules.fcn.head module hat.models.task_modules.fcn.target module hat.models.task_modules.fcos.decoder module hat.models.task_modules.fcos.fcos_loss module hat.models.task_modules.fcos.filter module hat.models.task_modules.fcos.head module hat.models.task_modules.fcos.target module hat.models.task_modules.fcos3d.bbox_coder module hat.models.task_modules.fcos3d.loss module hat.models.task_modules.fcos3d.post_process module hat.models.task_modules.fcos3d.target module hat.models.task_modules.ganet.decoder module hat.models.task_modules.ganet.head module hat.models.task_modules.ganet.losses module hat.models.task_modules.ganet.neck module hat.models.task_modules.ganet.target module hat.models.task_modules.lidar.anchor_generator module hat.models.task_modules.lidar.box_coders module hat.models.task_modules.lidar.pillar_encoder module hat.models.task_modules.lidar.target_assigner module hat.models.task_modules.lidar_multitask.decoder module hat.models.task_modules.motion_forecasting.decoders.densetnt.head module hat.models.task_modules.motion_forecasting.decoders.densetnt.loss module hat.models.task_modules.motion_forecasting.decoders.densetnt.post_process module hat.models.task_modules.motion_forecasting.decoders.densetnt.target module hat.models.task_modules.motion_forecasting.encoders.vectornet module hat.models.task_modules.motr.criterion module hat.models.task_modules.motr.head module hat.models.task_modules.motr.motr_deformable_transformer module hat.models.task_modules.motr.post_process module hat.models.task_modules.motr.qim module hat.models.task_modules.petr.head module hat.models.task_modules.pointpillars.head module hat.models.task_modules.pointpillars.loss module hat.models.task_modules.pointpillars.postprocess module hat.models.task_modules.pointpillars.preprocess module hat.models.task_modules.pwcnet.head module hat.models.task_modules.pwcnet.neck module hat.models.task_modules.retinanet.filter module hat.models.task_modules.retinanet.head module hat.models.task_modules.retinanet.postprocess module hat.models.task_modules.seg.decoder module hat.models.task_modules.seg.head module hat.models.task_modules.seg.target module hat.models.task_modules.seg.utils module hat.models.task_modules.seg.vargnet_seg_head module hat.models.task_modules.stereonet.head module hat.models.task_modules.stereonet.headplus module hat.models.task_modules.stereonet.neck module hat.models.task_modules.stereonet.post_process module hat.models.task_modules.view_fusion.cft_transformer module hat.models.task_modules.view_fusion.decoder module hat.models.task_modules.view_fusion.encoder module hat.models.task_modules.view_fusion.temporal_fusion module hat.models.task_modules.view_fusion.view_transformer module hat.models.task_modules.yolo.anchor module hat.models.task_modules.yolo.filter module hat.models.task_modules.yolo.head module hat.models.task_modules.yolo.label_encoder module hat.models.task_modules.yolo.matcher module hat.models.task_modules.yolo.postprocess module hat.profiler.memory_profiler module hat.profiler.model_profiler module hat.profiler.profilers module hat.visualize.argoverse module hat.visualize.cam3d module hat.visualize.cls module hat.visualize.det module hat.visualize.disparity module hat.visualize.keypoints module hat.visualize.lane_lines module hat.visualize.nuscenes module hat.visualize.opticalflow module hat.visualize.seg module hat.visualize.track module HeatMap3DTargetGenerator (class in hat.models.base_modules.target.heatmap_roi_3d_target) HeatmapDecoder (class in hat.models.task_modules.carfusion_keypoints.heatmap_decoder) HeatmapKeypointModel (class in hat.models.structures.keypoints.keypoint_model) horizon_plugin_pytorch.functional module, [1], [2] horizon_plugin_pytorch.march module horizon_plugin_pytorch.nn.bgr_to_yuv444 module horizon_plugin_pytorch.nn.detection_post_process module horizon_plugin_pytorch.nn.detection_post_process_v1 module horizon_plugin_pytorch.nn.functional module horizon_plugin_pytorch.quantization module horizon_plugin_pytorch.quantization.fake_quantize module horizon_plugin_pytorch.quantization.fuse_modules module, [1] horizon_plugin_pytorch.quantization.observer module horizon_plugin_pytorch.quantization.qconfig module horizon_plugin_pytorch.quantization.quantize module horizon_plugin_pytorch.quantization.quantize_fx module horizon_plugin_pytorch.utils.fx_helper module HorizonAdasClsPostProcessor (class in hat.models.base_modules.postprocess.argmax_postprocess) HorizonSwinTransformer (class in hat.models.backbones.horizon_swin_transformer) HueSaturationValue (class in hat.data.transforms.detection)

I

IgRegionMatcher (class in hat.models.base_modules.matcher) ImageNet (class in hat.data.datasets.imagenet) ImageNetFromImage (class in hat.data.datasets.imagenet) img_encode() (hat.models.structures.view_fusion.ViewFusion method) init_weights() (hat.models.backbones.horizon_swin_transformer.HorizonSwinTransformer method) (hat.models.base_modules.transformer_attentions.BevDeformableTemporalAttention method) (hat.models.base_modules.transformer_attentions.BevSpatialCrossAtten method) (hat.models.base_modules.transformer_attentions.MSDeformableAttention3D method) (hat.models.base_modules.transformer_attentions.ObjectDetr3DCrossAtten method) (hat.models.necks.retinanet_fpn.RetinaNetFPN method) (hat.models.task_modules.detr.transformer.Transformer method) (hat.models.task_modules.pwcnet.head.PwcNetHead method) (hat.models.task_modules.pwcnet.neck.PwcNetNeck method) (hat.models.task_modules.retinanet.head.RetinaNetHead method) (hat.models.task_modules.stereonet.head.StereoNetHead method) (hat.models.task_modules.stereonet.neck.StereoNetNeck method)

inverse_transform() (hat.data.transforms.detection.FixedCrop method) (hat.data.transforms.detection.PresetCrop method) (hat.data.transforms.detection.Resize method) (hat.data.transforms.segmentation.SegResizeAffine method) IterableDetRoIListTransform (class in hat.data.transforms.detection) IterableDetRoITransform (class in hat.data.transforms.detection)

J

JPEGCompress (class in hat.data.transforms.faceid)

K

KeypointsViz (class in hat.visualize.keypoints) Kitti2D (class in hat.data.datasets.kitti2d) Kitti2DMetric (class in hat.metrics.kitti2d_detection)

Kitti3D (class in hat.data.datasets.kitti3d) Kitti3DDetection (class in hat.data.datasets.kitti3d) Kitti3DMetricDet (class in hat.metrics.kitti3d_detection)

L

L1Loss (class in hat.models.losses.l1_loss) LabelRemap (class in hat.data.transforms.segmentation) LabelSmooth (class in hat.data.transforms.classification) LaneFastFocalLoss (class in hat.models.losses.focal_loss) LanelineViz (class in hat.visualize.lane_lines) LearnedPositionalEncoding (class in hat.models.base_modules.position_encoding) LidarDetDecoder (class in hat.models.task_modules.lidar_multitask.decoder) LidarMultiPreprocess (class in hat.data.transforms.lidar_utils.lidar_transform_3d) LidarMultiTask (class in hat.models.structures.lidar_multitask.lidar_multitask) LidarReformat (class in hat.data.transforms.lidar_utils.lidar_transform_3d) LidarSegDecoder (class in hat.models.task_modules.lidar_multitask.decoder) LidarTargetAssigner (class in hat.models.task_modules.lidar.target_assigner)

ListToDict (class in hat.data.transforms.common) LnNormLoss (class in hat.models.losses.lnnorm_loss) LoadCheckpoint (class in hat.models.model_convert.converters) LoadEMAModel (class in hat.callbacks.exponential_moving_average) LoadMeanTeacherCheckpoint (class in hat.models.model_convert.converters) loss_boxes() (hat.models.task_modules.detr.criterion.DetrCriterion method) (hat.models.task_modules.motr.criterion.MotrCriterion method) loss_cardinality() (hat.models.task_modules.detr.criterion.DetrCriterion method) loss_labels() (hat.models.task_modules.detr.criterion.DetrCriterion method) (hat.models.task_modules.motr.criterion.MotrCriterion method) LossShow (class in hat.metrics.loss_show) LSSTransformer (class in hat.models.task_modules.view_fusion.view_transformer)

M

March (class in horizon_plugin_pytorch.march) match_thresholds (hat.models.task_modules.lidar.anchor_generator.Anchor3DGeneratorStride property) MatchLabelGroundLineEncoder (class in hat.models.base_modules.label_encoder) MatchLabelSepEncoder (class in hat.models.base_modules.label_encoder) MatchLabelTrackEncoder (class in hat.models.base_modules.label_encoder) MaxIoUMatcher (class in hat.models.base_modules.matcher) MaxPostProcess (class in hat.models.base_modules.postprocess.max_postprocess) MeanBlur (class in hat.data.transforms.detection) MeanIOU (class in hat.metrics.mean_iou) MeanKeypointDist (class in hat.metrics.metric_keypoints) MeanTeacherValidation (class in hat.callbacks.validation) MedianBlur (class in hat.data.transforms.detection) MetricUpdater (class in hat.callbacks.metric_updater) MinIoURandomCrop (class in hat.data.transforms.detection) mix_img_transform() (hat.data.transforms.detection.DetMosaic method) (hat.data.transforms.detection.DetYOLOv5MixUp method) MixSegLoss (class in hat.models.losses.seg_loss) MixSegLossMultipreds (class in hat.models.losses.seg_loss) MixVarGENet (class in hat.models.backbones.mixvargenet) MobileNetV1 (class in hat.models.backbones.mobilenetv1) MobileNetV2 (class in hat.models.backbones.mobilenetv2) module hat.callbacks.checkpoint hat.callbacks.dump_data hat.callbacks.exponential_moving_average hat.callbacks.freeze_bn hat.callbacks.grad_scale hat.callbacks.lr_updater hat.callbacks.metric_updater hat.callbacks.monitor hat.callbacks.online_model_trick hat.callbacks.param_updater hat.callbacks.save_traced hat.callbacks.tensorboard hat.callbacks.validation hat.data.collates.collates hat.data.dataloaders.passthrough_dataloader hat.data.datasets.argoverse_dataset hat.data.datasets.batch_transform_dataset hat.data.datasets.carfusion_keypoints_dataset hat.data.datasets.cityscapes hat.data.datasets.culane_dataset hat.data.datasets.dataset_wrappers hat.data.datasets.flyingchairs_dataset hat.data.datasets.imagenet hat.data.datasets.kitti2d hat.data.datasets.kitti3d hat.data.datasets.mot17_dataset hat.data.datasets.mscoco hat.data.datasets.nuscenes_dataset hat.data.datasets.rand_dataset hat.data.datasets.sceneflow_dataset hat.data.datasets.voc hat.data.samplers.dist_cycle_sampler_multi_dataset hat.data.samplers.dist_group_sampler hat.data.samplers.dist_sampler hat.data.samplers.dist_set_epoch_dataset_sampler hat.data.samplers.selected_sampler hat.data.transforms.classification hat.data.transforms.common hat.data.transforms.detection hat.data.transforms.faceid hat.data.transforms.gaze.gaze hat.data.transforms.grid_mask hat.data.transforms.keypoints hat.data.transforms.lidar hat.data.transforms.lidar_utils.lidar_transform_3d hat.data.transforms.lidar_utils.preprocess hat.data.transforms.lidar_utils.sample_ops hat.data.transforms.multi_views hat.data.transforms.segmentation hat.data.transforms.seq_transform hat.engine.calibrator hat.engine.ddp_trainer hat.engine.dp_trainer hat.engine.predictor hat.engine.processors.processor hat.engine.trainer hat.metrics.acc hat.metrics.argoverse_metric hat.metrics.coco_detection hat.metrics.kitti2d_detection hat.metrics.kitti3d_detection hat.metrics.loss_show hat.metrics.mean_iou hat.metrics.metric_keypoints hat.metrics.metric_lane_detection hat.metrics.metric_optical_flow hat.metrics.mot_metrics hat.metrics.nuscenes_metric hat.metrics.voc_detection hat.models.backbones.contrib.resnet hat.models.backbones.efficientnet hat.models.backbones.horizon_swin_transformer hat.models.backbones.mixvargenet hat.models.backbones.mobilenetv1 hat.models.backbones.mobilenetv2 hat.models.backbones.resnet hat.models.backbones.vargconvnet hat.models.backbones.vargdarknet hat.models.backbones.vargnetv2 hat.models.base_modules.basic_vargnet_module hat.models.base_modules.bbox_decoder hat.models.base_modules.dequant_module hat.models.base_modules.extend_container hat.models.base_modules.label_encoder hat.models.base_modules.matcher hat.models.base_modules.position_encoding hat.models.base_modules.postprocess.anchor_postprocess hat.models.base_modules.postprocess.argmax_postprocess hat.models.base_modules.postprocess.max_postprocess hat.models.base_modules.postprocess.rle_postprocess hat.models.base_modules.quant_module hat.models.base_modules.resize_parser hat.models.base_modules.roi_feat_extractors hat.models.base_modules.target.bbox_target hat.models.base_modules.target.heatmap_roi_3d_target hat.models.base_modules.target.reshape_target hat.models.base_modules.transformer_attentions hat.models.base_modules.transformer_bricks hat.models.embeddings hat.models.losses.cross_entropy_loss hat.models.losses.focal_loss hat.models.losses.giou_loss hat.models.losses.hinge_loss hat.models.losses.l1_loss hat.models.losses.lnnorm_loss hat.models.losses.mse_loss hat.models.losses.seg_loss hat.models.losses.smooth_l1_loss hat.models.losses.yolo_losses hat.models.model_convert.converters hat.models.model_convert.pipelines hat.models.necks.bifpn hat.models.necks.dw_unet hat.models.necks.fast_scnn hat.models.necks.fpn hat.models.necks.pafpn hat.models.necks.retinanet_fpn hat.models.necks.second_neck hat.models.necks.unet hat.models.necks.yolov3 hat.models.necks.yolov3_group hat.models.structures.classifier hat.models.structures.detectors.centerpoint hat.models.structures.detectors.detr hat.models.structures.detectors.detr3d hat.models.structures.detectors.fcos hat.models.structures.detectors.fcos3d hat.models.structures.detectors.pointpillars hat.models.structures.detectors.retinanet hat.models.structures.detectors.yolov3 hat.models.structures.disparity_pred.stereonet hat.models.structures.encoder_decoder hat.models.structures.keypoints.keypoint_model hat.models.structures.lane_pred.ganet hat.models.structures.lidar_multitask.lidar_multitask hat.models.structures.motion_forecasting hat.models.structures.opticalflow.pwcnet hat.models.structures.segmentor hat.models.structures.track_pred.motr hat.models.structures.view_fusion hat.models.task_modules.carfusion_keypoints.heatmap_decoder hat.models.task_modules.carfusion_keypoints.keypoint_head hat.models.task_modules.centerpoint.bbox_coders hat.models.task_modules.centerpoint.decoder hat.models.task_modules.centerpoint.head hat.models.task_modules.centerpoint.loss hat.models.task_modules.centerpoint.post_process hat.models.task_modules.centerpoint.pre_process hat.models.task_modules.centerpoint.target hat.models.task_modules.deeplab.head hat.models.task_modules.detr.criterion hat.models.task_modules.detr.head hat.models.task_modules.detr.post_process hat.models.task_modules.detr.transformer hat.models.task_modules.detr3d.head hat.models.task_modules.detr3d.post_process hat.models.task_modules.detr3d.target hat.models.task_modules.fcn.decoder hat.models.task_modules.fcn.head hat.models.task_modules.fcn.target hat.models.task_modules.fcos.decoder hat.models.task_modules.fcos.fcos_loss hat.models.task_modules.fcos.filter hat.models.task_modules.fcos.head hat.models.task_modules.fcos.target hat.models.task_modules.fcos3d.bbox_coder hat.models.task_modules.fcos3d.loss hat.models.task_modules.fcos3d.post_process hat.models.task_modules.fcos3d.target hat.models.task_modules.ganet.decoder hat.models.task_modules.ganet.head hat.models.task_modules.ganet.losses hat.models.task_modules.ganet.neck hat.models.task_modules.ganet.target hat.models.task_modules.lidar.anchor_generator hat.models.task_modules.lidar.box_coders hat.models.task_modules.lidar.pillar_encoder hat.models.task_modules.lidar.target_assigner hat.models.task_modules.lidar_multitask.decoder hat.models.task_modules.motion_forecasting.decoders.densetnt.head hat.models.task_modules.motion_forecasting.decoders.densetnt.loss hat.models.task_modules.motion_forecasting.decoders.densetnt.post_process hat.models.task_modules.motion_forecasting.decoders.densetnt.target hat.models.task_modules.motion_forecasting.encoders.vectornet hat.models.task_modules.motr.criterion hat.models.task_modules.motr.head hat.models.task_modules.motr.motr_deformable_transformer hat.models.task_modules.motr.post_process hat.models.task_modules.motr.qim hat.models.task_modules.petr.head hat.models.task_modules.pointpillars.head hat.models.task_modules.pointpillars.loss hat.models.task_modules.pointpillars.postprocess hat.models.task_modules.pointpillars.preprocess hat.models.task_modules.pwcnet.head hat.models.task_modules.pwcnet.neck hat.models.task_modules.retinanet.filter hat.models.task_modules.retinanet.head hat.models.task_modules.retinanet.postprocess hat.models.task_modules.seg.decoder hat.models.task_modules.seg.head hat.models.task_modules.seg.target hat.models.task_modules.seg.utils hat.models.task_modules.seg.vargnet_seg_head hat.models.task_modules.stereonet.head hat.models.task_modules.stereonet.headplus hat.models.task_modules.stereonet.neck hat.models.task_modules.stereonet.post_process hat.models.task_modules.view_fusion.cft_transformer hat.models.task_modules.view_fusion.decoder hat.models.task_modules.view_fusion.encoder hat.models.task_modules.view_fusion.temporal_fusion hat.models.task_modules.view_fusion.view_transformer hat.models.task_modules.yolo.anchor hat.models.task_modules.yolo.filter hat.models.task_modules.yolo.head hat.models.task_modules.yolo.label_encoder hat.models.task_modules.yolo.matcher hat.models.task_modules.yolo.postprocess hat.profiler.memory_profiler hat.profiler.model_profiler hat.profiler.profilers hat.visualize.argoverse hat.visualize.cam3d hat.visualize.cls hat.visualize.det hat.visualize.disparity hat.visualize.keypoints hat.visualize.lane_lines hat.visualize.nuscenes hat.visualize.opticalflow hat.visualize.seg hat.visualize.track horizon_plugin_pytorch.functional, [1], [2] horizon_plugin_pytorch.march horizon_plugin_pytorch.nn.bgr_to_yuv444 horizon_plugin_pytorch.nn.detection_post_process horizon_plugin_pytorch.nn.detection_post_process_v1 horizon_plugin_pytorch.nn.functional horizon_plugin_pytorch.quantization horizon_plugin_pytorch.quantization.fake_quantize horizon_plugin_pytorch.quantization.fuse_modules, [1] horizon_plugin_pytorch.quantization.observer horizon_plugin_pytorch.quantization.qconfig horizon_plugin_pytorch.quantization.quantize horizon_plugin_pytorch.quantization.quantize_fx horizon_plugin_pytorch.utils.fx_helper

Mosaic (class in hat.data.transforms.detection) Mot17Dataset (class in hat.data.datasets.mot17_dataset) Mot17FromImage (class in hat.data.datasets.mot17_dataset) Mot17Packer (class in hat.data.datasets.mot17_dataset) MotionBlur (class in hat.data.transforms.faceid) MotionForecasting (class in hat.models.structures.motion_forecasting) MotMetric (class in hat.metrics.mot_metrics) Motr (class in hat.models.structures.track_pred.motr) MotrCriterion (class in hat.models.task_modules.motr.criterion) MotrDeformableTransformer (class in hat.models.task_modules.motr.motr_deformable_transformer) MotrHead (class in hat.models.task_modules.motr.head) MotrPostProcess (class in hat.models.task_modules.motr.post_process) MovingAverageMinMaxObserver (class in horizon_plugin_pytorch.quantization.observer) MovingAveragePerChannelMinMaxObserver (class in horizon_plugin_pytorch.quantization.observer) MSDeformableAttention3D (class in hat.models.base_modules.transformer_attentions) MSELoss (class in hat.models.losses.mse_loss) MultiBatchProcessor (class in hat.engine.processors.processor) MultiClassMatchLabelSepEncoder (class in hat.models.base_modules.label_encoder) MultiScaleRoIAlign (class in hat.models.base_modules.roi_feat_extractors) MultiStageBatchProcessor (class in hat.engine.processors.processor) MultiStrideLosses (class in hat.models.losses.seg_loss) MultiTaskAnnoWrapper (class in hat.data.transforms.common) MultiViewsGridMask (class in hat.data.transforms.multi_views) MultiViewsImgCrop (class in hat.data.transforms.multi_views) MultiViewsImgFlip (class in hat.data.transforms.multi_views) MultiViewsImgResize (class in hat.data.transforms.multi_views) MultiViewsImgRotate (class in hat.data.transforms.multi_views) MultiViewsImgTransformWrapper (class in hat.data.transforms.multi_views) MultiViewsSpiltImgTransformWrapper (class in hat.data.transforms.multi_views)

N

nearest_iou_similarity() (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner method) nms() (hat.models.task_modules.pointpillars.postprocess.PointPillarsPostProcess method) NoamLrUpdater (class in hat.callbacks.lr_updater) Normalize (class in hat.data.transforms.detection) num_anchors_per_localization (hat.models.task_modules.lidar.anchor_generator.Anchor3DGeneratorStride property) num_anchors_per_location (hat.models.task_modules.lidar.target_assigner.LidarTargetAssigner property) num_of_anchor_sets (hat.models.task_modules.lidar.anchor_generator.Anchor3DGeneratorStride property) NuscenesBevDataset (class in hat.data.datasets.nuscenes_dataset) NuscenesBevSequenceDataset (class in hat.data.datasets.nuscenes_dataset)

NuscenesFromImage (class in hat.data.datasets.nuscenes_dataset) NuscenesFromImageSequence (class in hat.data.datasets.nuscenes_dataset) NuscenesLidarDataset (class in hat.data.datasets.nuscenes_dataset) NuscenesLidarWithSegDataset (class in hat.data.datasets.nuscenes_dataset) NuscenesMetric (class in hat.metrics.nuscenes_metric) NuscenesMonoDataset (class in hat.data.datasets.nuscenes_dataset) NuscenesMonoFromImage (class in hat.data.datasets.nuscenes_dataset) NuscenesMonoMetric (class in hat.metrics.nuscenes_metric) NuscenesViz (class in hat.visualize.nuscenes)

O

ObjectDetr3DCrossAtten (class in hat.models.base_modules.transformer_attentions) ObjectNoise (class in hat.data.transforms.lidar_utils.lidar_transform_3d) ObjectRangeFilter (class in hat.data.transforms.lidar_utils.lidar_transform_3d) ObjectSample (class in hat.data.transforms.lidar_utils.lidar_transform_3d) observer (horizon_plugin_pytorch.quantization.fake_quantize.FakeQuantize attribute) on_backward_end() (hat.callbacks.grad_scale.GradScale method) on_batch_end() (hat.callbacks.metric_updater.MetricUpdater method) (hat.callbacks.tensorboard.TensorBoard method) on_loop_begin() (hat.callbacks.lr_updater.CosineAnnealingLrUpdater method) (hat.callbacks.lr_updater.CosLrUpdater method) (hat.callbacks.lr_updater.CyclicLrUpdater method) (hat.callbacks.lr_updater.OneCycleUpdater method) (hat.callbacks.online_model_trick.FuseBN method) (hat.callbacks.param_updater.CosOptimParamUpdater method) (hat.callbacks.param_updater.CyclicOptimParamUpdater method)

one_hot_f() (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method) OneCycleUpdater (class in hat.callbacks.lr_updater) OneHot (class in hat.data.transforms.classification) OneHotClassEncoder (class in hat.models.base_modules.label_encoder)

P

pack_data() (hat.data.datasets.argoverse_dataset.Argoverse1Packer method) (hat.data.datasets.culane_dataset.CuLanePacker method) (hat.data.datasets.flyingchairs_dataset.FlyingChairsPacker method) (hat.data.datasets.mot17_dataset.Mot17Packer method) (hat.data.datasets.sceneflow_dataset.SceneFlowPacker method) PadTensorListToBatch (class in hat.data.transforms.detection) PAFPN (class in hat.models.necks.pafpn) ParsePointCloud (class in hat.data.transforms.lidar) PascalVOC (class in hat.data.datasets.voc) PassThroughDataLoader (class in hat.data.dataloaders.passthrough_dataloader) PassThroughProfiler (class in hat.profiler.profilers) PCKMetric (class in hat.metrics.metric_keypoints) perf_model() (in module horizon_plugin_pytorch.quantization) PersonPositionLabelFromMatch (class in hat.models.base_modules.label_encoder) PETRDecoder (class in hat.models.task_modules.petr.head) PETRHead (class in hat.models.task_modules.petr.head) PETRTransformer (class in hat.models.task_modules.petr.head) PillarFeatureNet (class in hat.models.task_modules.lidar.pillar_encoder) PILToTensor (class in hat.data.transforms.common) PlainCopyPaste (class in hat.data.transforms.detection) Point2VCS (class in hat.data.transforms.lidar) point_pillars_preprocess() (in module horizon_plugin_pytorch.nn.functional) PointCloudSegPreprocess (class in hat.data.transforms.lidar_utils.lidar_transform_3d) PointGlobalRotation (class in hat.data.transforms.lidar_utils.lidar_transform_3d) PointGlobalScaling (class in hat.data.transforms.lidar_utils.lidar_transform_3d) PointPillarScatter (class in hat.models.task_modules.lidar.pillar_encoder) PointPillarsDetector (class in hat.models.structures.detectors.pointpillars)

PointPillarsHead (class in hat.models.task_modules.pointpillars.head) PointPillarsLoss (class in hat.models.task_modules.pointpillars.loss) PointPillarsPostProcess (class in hat.models.task_modules.pointpillars.postprocess) PointPillarsPreProcess (class in hat.models.task_modules.pointpillars.preprocess) PointRandomFlip (class in hat.data.transforms.lidar_utils.lidar_transform_3d) PolyLrUpdater (class in hat.callbacks.lr_updater) position_embeding() (hat.models.task_modules.petr.head.PETRHead method) PositionEmbeddingLearned (class in hat.models.embeddings) PositionEmbeddingSine (class in hat.models.embeddings) Predictor (class in hat.engine.predictor) prepare_calibration() (in module horizon_plugin_pytorch.quantization.quantize) prepare_calibration_fx() (in module horizon_plugin_pytorch.quantization.quantize_fx) prepare_loss_weights() (hat.models.task_modules.pointpillars.loss.PointPillarsLoss method) prepare_qat() (in module horizon_plugin_pytorch.quantization.quantize) prepare_qat_fx() (in module horizon_plugin_pytorch.quantization.quantize_fx) PresetCrop (class in hat.data.transforms.detection) process() (hat.callbacks.online_model_trick.FreezeModule method) (hat.callbacks.online_model_trick.FuseBN method) process_sequence_input() (hat.models.backbones.mixvargenet.MixVarGENet method) (hat.models.backbones.vargnetv2.VargNetV2 method) profile() (hat.profiler.memory_profiler.StageCPUMemoryProfiler method) ProfileFeaturemap (class in hat.profiler.model_profiler) ProposalTarget (class in hat.models.base_modules.target.bbox_target) ProposalTargetGroundLine (class in hat.models.base_modules.target.bbox_target) ProposalTargetTrack (class in hat.models.base_modules.target.bbox_target) PwcNet (class in hat.models.structures.opticalflow.pwcnet) PwcNetHead (class in hat.models.task_modules.pwcnet.head) PwcNetNeck (class in hat.models.task_modules.pwcnet.neck)

Q

QATFuseBNConvertPipeline (class in hat.models.model_convert.pipelines) QATFusePartBN (class in hat.models.model_convert.converters)

quantized_conv2d() (in module horizon_plugin_pytorch.functional) QuantModule (class in hat.models.base_modules.quant_module) QueryInteractionModule (class in hat.models.task_modules.motr.qim)

R

RandDataset (class in hat.data.datasets.rand_dataset) RandomBrightnessContrast (class in hat.data.transforms.detection) RandomColorJitter (class in hat.data.transforms.gaze.gaze) RandomDownSample (class in hat.data.transforms.faceid) RandomExpand (class in hat.data.transforms.detection) RandomFlip (class in hat.data.transforms.detection) RandomGray (class in hat.data.transforms.faceid) RandomPadLdmkData (class in hat.data.transforms.keypoints) RandomResizedCrop (class in hat.data.transforms.detection) RandomSelectOne (class in hat.data.transforms.common) RCNN3DLabelFromMatch (class in hat.models.base_modules.label_encoder) RCNNBinDetLabelFromMatch (class in hat.models.base_modules.label_encoder) RCNNKPSLabelFromMatch (class in hat.models.base_modules.label_encoder) RenameKeys (class in hat.data.transforms.common) RepeatDataset (class in hat.data.datasets.dataset_wrappers) RepModel2Deploy (class in hat.models.model_convert.converters) ResampleDataset (class in hat.data.datasets.dataset_wrappers) reset() (hat.metrics.argoverse_metric.ArgoverseMetric method) (hat.metrics.coco_detection.COCODetectionMetric method) (hat.metrics.kitti2d_detection.Kitti2DMetric method) (hat.metrics.kitti3d_detection.Kitti3DMetricDet method) (hat.metrics.loss_show.LossShow method) (hat.metrics.mot_metrics.MotMetric method) (hat.metrics.nuscenes_metric.NuscenesMetric method) (hat.metrics.voc_detection.VOCMApMetric method)

ReshapeTarget (class in hat.models.base_modules.target.reshape_target) Resize (class in hat.data.transforms.detection) Resize3D (class in hat.data.transforms.detection) ResizeParser (class in hat.models.base_modules.resize_parser) ResNet18 (class in hat.models.backbones.resnet) resnet18() (in module hat.models.backbones.contrib.resnet) ResNet50 (class in hat.models.backbones.resnet) resnet50() (in module hat.models.backbones.contrib.resnet) ResNet50V2 (class in hat.models.backbones.resnet) ResNetV2 (class in hat.models.backbones.resnet) RetinaNet (class in hat.models.structures.detectors.retinanet) RetinaNetFPN (class in hat.models.necks.retinanet_fpn) RetinaNetHead (class in hat.models.task_modules.retinanet.head) RetinanetMultiStrideFilter (class in hat.models.task_modules.retinanet.filter) RetinaNetPostProcess (class in hat.models.task_modules.retinanet.postprocess) RGBShift (class in hat.data.transforms.detection) RLEPostprocess (class in hat.models.base_modules.postprocess.rle_postprocess)

S

save_snapshots() (hat.profiler.memory_profiler.GPUMemoryProfiler method) SaveTraced (class in hat.callbacks.save_traced) Scale (class in hat.data.transforms.segmentation) SceneFlow (class in hat.data.datasets.sceneflow_dataset) SceneFlowFromImage (class in hat.data.datasets.sceneflow_dataset) SceneFlowPacker (class in hat.data.datasets.sceneflow_dataset) SECONDNeck (class in hat.models.necks.second_neck) SegDecoder (class in hat.models.task_modules.seg.decoder) SegEdgeLoss (class in hat.models.losses.seg_loss) SegHead (class in hat.models.task_modules.seg.head) SegLoss (class in hat.models.losses.seg_loss) Segmentor (class in hat.models.structures.segmentor) SegmentorV2 (class in hat.models.structures.segmentor) SegOneHot (class in hat.data.transforms.segmentation) SegRandomAffine (class in hat.data.transforms.segmentation) SegRandomCrop (class in hat.data.transforms.segmentation) SegRandomCutOut (class in hat.data.transforms.segmentation) SegResize (class in hat.data.transforms.segmentation) SegResizeAffine (class in hat.data.transforms.segmentation) SegReWeightByArea (class in hat.data.transforms.segmentation) SegTarget (class in hat.models.task_modules.seg.target) SegViz (class in hat.visualize.seg) select_goals_by_NMS() (hat.models.task_modules.motion_forecasting.decoders.densetnt.post_process.DensetntPostprocess method) SelectedSampler (class in hat.data.samplers.selected_sampler) SeqAlbuImageOnlyTransform (class in hat.data.transforms.seq_transform) SeqAugmentHSV (class in hat.data.transforms.seq_transform) SeqBgrToYuv444 (class in hat.data.transforms.seq_transform) SeqNormalize (class in hat.data.transforms.seq_transform) SeqRandomFlip (class in hat.data.transforms.seq_transform) SeqRandomSizeCrop (class in hat.data.transforms.seq_transform) SeqResize (class in hat.data.transforms.seq_transform) SeqToFasterRCNNData (class in hat.data.transforms.seq_transform) SeqToTensor (class in hat.data.transforms.seq_transform) set_calibration_qconfig() (hat.models.structures.detectors.centerpoint.CenterPointDetector method) (hat.models.structures.detectors.detr3d.Detr3d method) (hat.models.structures.detectors.pointpillars.PointPillarsDetector method) (hat.models.task_modules.petr.head.PETRHead method) set_epoch() (hat.data.samplers.dist_group_sampler.DistributedGroupSampler method) (hat.data.samplers.dist_set_epoch_dataset_sampler.DistSetEpochDatasetSampler method) (hat.data.samplers.selected_sampler.SelectedSampler method) set_qconfig() (hat.models.structures.detectors.centerpoint.CenterPointDetector method) (hat.models.structures.detectors.detr3d.Detr3d method) (hat.models.structures.detectors.pointpillars.PointPillarsDetector method) (hat.models.structures.disparity_pred.stereonet.StereoNet method) (hat.models.structures.lidar_multitask.lidar_multitask.LidarMultiTask method) (hat.models.structures.view_fusion.ViewFusion method) (hat.models.task_modules.centerpoint.head.CenterPointHead method) (hat.models.task_modules.detr3d.head.Detr3dHead method) (hat.models.task_modules.detr3d.head.Detr3dTransformer method) (hat.models.task_modules.motion_forecasting.decoders.densetnt.head.Densetnt method) (hat.models.task_modules.motion_forecasting.encoders.vectornet.Vectornet method) (hat.models.task_modules.petr.head.PETRDecoder method) (hat.models.task_modules.petr.head.PETRHead method) (hat.models.task_modules.petr.head.PETRTransformer method) (hat.models.task_modules.stereonet.head.StereoNetHead method) (hat.models.task_modules.stereonet.headplus.StereoNetHeadPlus method) (hat.models.task_modules.stereonet.neck.StereoNetNeck method) (hat.models.task_modules.view_fusion.cft_transformer.CFTAuxHead method) (hat.models.task_modules.view_fusion.cft_transformer.CFTTransformer method) (hat.models.task_modules.view_fusion.encoder.BevEncoder method) (hat.models.task_modules.view_fusion.view_transformer.LSSTransformer method)

set_qparams() (horizon_plugin_pytorch.quantization.fake_quantize.FakeQuantize method) set_warmup_training_lr() (hat.callbacks.lr_updater.OneCycleUpdater method) ShiftScaleRotate (class in hat.data.transforms.detection) ShufflePoints (class in hat.data.transforms.lidar_utils.lidar_transform_3d) SimpleDataset (class in hat.data.datasets.rand_dataset) SimpleProfiler (class in hat.profiler.profilers) SmoothL1Loss (class in hat.models.losses.smooth_l1_loss) SoftmaxFocalLoss (class in hat.models.losses.focal_loss) SoftTargetCrossEntropy (class in hat.models.losses.cross_entropy_loss) SpatialVariantBrightness (class in hat.data.transforms.faceid) StageCPUMemoryProfiler (class in hat.profiler.memory_profiler) start() (hat.profiler.memory_profiler.CPUMemoryProfiler method) (hat.profiler.memory_profiler.GPUMemoryProfiler method) (hat.profiler.profilers.PassThroughProfiler method) (hat.profiler.profilers.SimpleProfiler method) StatsMonitor (class in hat.callbacks.monitor) StepDecayLrUpdater (class in hat.callbacks.lr_updater) StepDecayOptimParamUpdater (class in hat.callbacks.param_updater) StereoNet (class in hat.models.structures.disparity_pred.stereonet) StereoNetHead (class in hat.models.task_modules.stereonet.head) StereoNetHeadPlus (class in hat.models.task_modules.stereonet.headplus) StereoNetNeck (class in hat.models.task_modules.stereonet.neck) StereoNetPlus (class in hat.models.structures.disparity_pred.stereonet) StereoNetPostProcess (class in hat.models.task_modules.stereonet.post_process) StereoNetPostProcessPlus (class in hat.models.task_modules.stereonet.post_process) stop() (hat.profiler.memory_profiler.CPUMemoryProfiler method) (hat.profiler.memory_profiler.GPUMemoryProfiler method) (hat.profiler.profilers.PassThroughProfiler method) (hat.profiler.profilers.SimpleProfiler method) summary() (hat.profiler.memory_profiler.CPUMemoryProfiler method) (hat.profiler.memory_profiler.GPUMemoryProfiler method) (hat.profiler.profilers.PassThroughProfiler method) (hat.profiler.profilers.SimpleProfiler method)

T

TensorBoard (class in hat.callbacks.tensorboard) TensorToNumpy (class in hat.data.transforms.common) TimmMixup (class in hat.data.transforms.classification) TimmTransforms (class in hat.data.transforms.classification) TinyVargNetV2 (class in hat.models.backbones.vargnetv2) ToCUDA (class in hat.data.transforms.common) ToFasterRCNNData (class in hat.data.transforms.detection) ToLdmkRCNNData (class in hat.data.transforms.detection)

ToMultiTaskFasterRCNNData (class in hat.data.transforms.detection) TopKAccuracy (class in hat.metrics.acc) ToPositionFasterRCNNData (class in hat.data.transforms.detection) TorchCompile (class in hat.models.model_convert.converters) ToTensor (class in hat.data.transforms.detection) TrackViz (class in hat.visualize.track) Trainer (class in hat.engine.trainer) Transformer (class in hat.models.task_modules.detr.transformer) TransformerLayerSequence (class in hat.models.base_modules.transformer_bricks)

U

Undistortion (class in hat.data.transforms.common) Unet (class in hat.models.necks.unet) unmatch_thresholds (hat.models.task_modules.lidar.anchor_generator.Anchor3DGeneratorStride property) update() (hat.metrics.acc.Accuracy method) (hat.metrics.acc.AccuracyAttrMultiLabel method) (hat.metrics.acc.AccuracySeg method) (hat.metrics.acc.TopKAccuracy method) (hat.metrics.argoverse_metric.ArgoverseMetric method) (hat.metrics.coco_detection.COCODetectionMetric method) (hat.metrics.kitti2d_detection.Kitti2DMetric method) (hat.metrics.kitti3d_detection.Kitti3DMetricDet method) (hat.metrics.loss_show.LossShow method) (hat.metrics.mean_iou.MeanIOU method) (hat.metrics.metric_keypoints.MeanKeypointDist method) (hat.metrics.metric_keypoints.PCKMetric method) (hat.metrics.metric_lane_detection.CulaneF1Score method) (hat.metrics.metric_optical_flow.EndPointError method) (hat.metrics.mot_metrics.MotMetric method) (hat.metrics.nuscenes_metric.NuscenesMetric method) (hat.metrics.nuscenes_metric.NuscenesMonoMetric method) (hat.metrics.voc_detection.VOCMApMetric method)

V

Validation (class in hat.callbacks.validation) VargBevBackbone (class in hat.models.task_modules.view_fusion.encoder) VargCenterPointHead (class in hat.models.task_modules.centerpoint.head) VargConvNet (class in hat.models.backbones.vargconvnet) VarGDarkNet53 (class in hat.models.backbones.vargdarknet) VargNetSegDecoder (class in hat.models.task_modules.seg.decoder) VargNetV2 (class in hat.models.backbones.vargnetv2) VargPAFPN (class in hat.models.necks.pafpn) Vectornet (class in hat.models.task_modules.motion_forecasting.encoders.vectornet)

VehicleSideFCOSDecoder (class in hat.models.task_modules.fcos.decoder) VehicleSideFCOSHead (class in hat.models.task_modules.fcos.head) VehicleSideFCOSLoss (class in hat.models.task_modules.fcos.fcos_loss) VehicleSideFCOSTarget (class in hat.models.task_modules.fcos.target) ViewFusion (class in hat.models.structures.view_fusion) visualize_model() (in module horizon_plugin_pytorch.quantization) VOC07MApMetric (class in hat.metrics.voc_detection) VOCFromImage (class in hat.data.datasets.voc) VOCMApMetric (class in hat.metrics.voc_detection) Voxelization (class in hat.data.transforms.lidar)

W

warp() (hat.models.task_modules.pwcnet.head.PwcNetHead method) WeightedSquaredHingeLoss (class in hat.models.losses.hinge_loss)

with_args() (horizon_plugin_pytorch.quantization.fake_quantize.FakeQuantize class method) wrap() (in module horizon_plugin_pytorch.utils.fx_helper) WrappingTransformer (class in hat.models.task_modules.view_fusion.view_transformer)

X

XYWHBBoxDecoder (class in hat.models.base_modules.bbox_decoder)

XYWHBBoxEncoder (class in hat.models.base_modules.label_encoder)

Y

YoloGroupNeck (class in hat.models.necks.yolov3_group) YOLOV3 (class in hat.models.structures.detectors.yolov3) YOLOV3AnchorGenerator (class in hat.models.task_modules.yolo.anchor) YOLOv3Filter (class in hat.models.task_modules.yolo.filter) YOLOV3Head (class in hat.models.task_modules.yolo.head)

YOLOV3LabelEncoder (class in hat.models.task_modules.yolo.label_encoder) YOLOV3Loss (class in hat.models.losses.yolo_losses) YOLOV3Matcher (class in hat.models.task_modules.yolo.matcher) YOLOV3Neck (class in hat.models.necks.yolov3) YOLOV3PostProcess (class in hat.models.task_modules.yolo.postprocess)