Functions — github.com/Nota-NetsPresso/netspresso-trainer

hub / github.com/Nota-NetsPresso/netspresso-trainer / functions

Functions1,065 in github.com/Nota-NetsPresso/netspresso-trainer

⨍ Functions 1,065◇ Types & classes 251

Methodforward

(self, pred, target)

src/netspresso_trainer/losses/detection/yolox.py:471

Methodforward

(self, outputs, targets)

src/netspresso_trainer/losses/detection/rtdetr.py:89

Methodforward

This performs the loss computation. Parameters: outputs: dict of tensors, see the output specification of the model for the form

src/netspresso_trainer/losses/detection/rtdetr.py:276

Methodforward

(self, out: Dict, target: Dict)

src/netspresso_trainer/losses/detection/retinanet.py:49

Methodforward

(self, out, target, matched_idxs)

src/netspresso_trainer/losses/detection/retinanet.py:75

Methodforward

(self, out, target, matched_idxs)

src/netspresso_trainer/losses/detection/retinanet.py:114

Methodforward

(self, out: Dict, target: Dict)

src/netspresso_trainer/losses/segmentation/seg_cross_entropy.py:16

Methodforward

(self, out: Dict, target: Dict)

src/netspresso_trainer/losses/segmentation/pidnet.py:48

Methodforward

(self, out: Dict, target: Dict)

src/netspresso_trainer/losses/segmentation/pidnet.py:147

Methodforward

(self, out: Dict, target: Dict)

src/netspresso_trainer/losses/segmentation/pidnet.py:161

Functionfpn

(intermediate_features_dim, conf_model_neck, **kwargs)

src/netspresso_trainer/models/necks/experimental/fpn.py:169

Functiongelan

(task, conf_model_backbone)

src/netspresso_trainer/models/backbones/experimental/gelan.py:139

Functionget_augmented_images

(phase, task, model_name, yaml_str, test_image, num_samples=NUM_AUGMENTATION_SAMPLES)

demo/func/augmentation.py:19

Methodget_lr

(self)

src/netspresso_trainer/schedulers/cosine_lr.py:51

Methodget_lr

(self)

src/netspresso_trainer/schedulers/poly_lr.py:47

Methodget_lr

(self)

src/netspresso_trainer/schedulers/step_lr.py:65

Methodget_lr

(self)

src/netspresso_trainer/schedulers/multi_step_lr.py:52

Functionget_lr_dataframe_from_config

(yaml_str: str)

demo/func/scheduler.py:24

Methodget_metric_with_all_outputs

(self, outputs, phase: Literal['train', 'valid'], metric_factory)

src/netspresso_trainer/pipelines/task_processors/classification.py:176

Methodget_metric_with_all_outputs

(self, outputs, phase: Literal['train', 'valid'], metric_factory)

src/netspresso_trainer/pipelines/task_processors/detection.py:195

Methodget_metric_with_all_outputs

(self, outputs, phase: Literal['train', 'valid'], metric_factory)

src/netspresso_trainer/pipelines/task_processors/pose_estimation.py:155

Methodget_metric_with_all_outputs

(self, outputs, phase: Literal['train', 'valid'], metric_factory)

src/netspresso_trainer/pipelines/task_processors/segmentation.py:177

Methodget_name

Get the name of the model.

src/netspresso_trainer/models/base.py:210

Methodget_predictions

(self, results, class_map)

src/netspresso_trainer/pipelines/task_processors/classification.py:179

Methodget_predictions

(self, results, class_map)

src/netspresso_trainer/pipelines/task_processors/detection.py:201

Methodget_predictions

(self, results, class_map)

src/netspresso_trainer/pipelines/task_processors/pose_estimation.py:161

Methodget_predictions

(self, results, class_map)

src/netspresso_trainer/pipelines/task_processors/segmentation.py:180

Methodhead_list

(self)

src/netspresso_trainer/models/base.py:55

Methodhead_list

(self)

src/netspresso_trainer/models/full/experimental/pidnet.py:157

Functionid_from_dataframe_select

(df: pd.DataFrame, evt: gr.SelectData)

demo/tab/experiments.py:22

Methodinit_bn

(M)

src/netspresso_trainer/models/necks/experimental/yolov9fpn.py:153

Methodinit_bn

(M)

src/netspresso_trainer/models/necks/experimental/yolopafpn.py:118

Methodinit_bn

(M)

src/netspresso_trainer/models/backbones/experimental/darknet.py:141

Methodinit_bn

(M)

src/netspresso_trainer/models/backbones/experimental/gelan.py:102

Methodinit_bn

(M)

src/netspresso_trainer/models/backbones/experimental/mobilenetv3.py:126

Methodinit_bn

(M)

src/netspresso_trainer/models/heads/detection/experimental/anchor_free_decoupled_head.py:133

Methodinit_bn

(M)

src/netspresso_trainer/models/heads/detection/experimental/yolo_head.py:73

Functioninit_worker

(worker_id, worker_seeding='all')

src/netspresso_trainer/dataloaders/utils/loader.py:67

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/necks/experimental/yolov9fpn.py:205

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/necks/experimental/fpn.py:165

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/necks/experimental/fpn.py:202

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/necks/experimental/yolopafpn.py:158

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/necks/experimental/rtdetr_hybrid_encoder.py:173

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/backbones/core/resnet.py:201

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/backbones/experimental/darknet.py:175

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/backbones/experimental/gelan.py:132

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/backbones/experimental/mobilenetv3.py:155

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/backbones/experimental/mobilenetv4.py:195

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/backbones/experimental/shufflenetv2.py:102

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/backbones/experimental/mixnet.py:285

Methodintermediate_features_dim

(self)

src/netspresso_trainer/models/op/base_metaformer.py:471

Functionis_file_dict

(image_dir: Union[Path, str], file_or_dir_to_idx)

src/netspresso_trainer/dataloaders/utils/misc.py:78

Methodis_verified

(self)

demo/func/pynetspresso.py:20

Methodlearning_rate

(self)

src/netspresso_trainer/pipelines/train.py:146

Functionlightfpn

(intermediate_features_dim, conf_model_neck, **kwargs)

src/netspresso_trainer/models/necks/experimental/fpn.py:205

Functionload_augmentation_config_both

(task_choices)

demo/tab/home/main.py:43

Methodload_class_map

(self, id_mapping)

src/netspresso_trainer/dataloaders/classification.py:86

Methodload_class_map

(self, id_mapping)

src/netspresso_trainer/dataloaders/detection.py:91

Methodload_class_map

(self, id_mapping)

src/netspresso_trainer/dataloaders/pose_estimation.py:83

Methodload_data

(self, split='train')

src/netspresso_trainer/dataloaders/classification.py:43

Methodload_data

(self, split='train')

src/netspresso_trainer/dataloaders/detection.py:42

Methodload_data

(self, split='train')

src/netspresso_trainer/dataloaders/pose_estimation.py:39

Methodload_data

(self, split='train')

src/netspresso_trainer/dataloaders/segmentation.py:42

Methodload_huggingface_samples

(self)

src/netspresso_trainer/dataloaders/classification.py:90

Methodload_huggingface_samples

(self)

src/netspresso_trainer/dataloaders/detection.py:95

Methodload_huggingface_samples

(self)

src/netspresso_trainer/dataloaders/pose_estimation.py:87

Methodload_huggingface_samples

(self)

src/netspresso_trainer/dataloaders/segmentation.py:109

Methodload_id_mapping

(self)

src/netspresso_trainer/dataloaders/classification.py:68

Methodload_id_mapping

(self)

src/netspresso_trainer/dataloaders/detection.py:73

Methodload_id_mapping

(self)

src/netspresso_trainer/dataloaders/pose_estimation.py:70

Methodload_id_mapping

(self)

src/netspresso_trainer/dataloaders/segmentation.py:69

Methodlog_hparams

(self, hp_omegaconf: Union[Dict, List], final_metrics=None)

src/netspresso_trainer/loggers/tensorboard.py:102

Methodlog_image

(self, key, value: Union[np.ndarray, torch.Tensor], mode='train')

src/netspresso_trainer/loggers/tensorboard.py:77

Functionlogin_with_session

(session: NetsPressoSession, email: str, password: str)

demo/func/pynetspresso.py:72

Methodloss_boxes

Compute the losses related to the bounding boxes, the L1 regression loss and the GIoU loss targets dicts must contain the key "boxes" conta

src/netspresso_trainer/losses/detection/rtdetr.py:233

Methodloss_cardinality

Compute the cardinality error, ie the absolute error in the number of predicted non-empty boxes This is not really a loss, it is intended for

src/netspresso_trainer/losses/detection/rtdetr.py:220

Methodloss_labels

Classification loss (NLL) targets dicts must contain the key "labels" containing a tensor of dim [nb_target_boxes]

src/netspresso_trainer/losses/detection/rtdetr.py:155

Methodloss_labels_focal

(self, outputs, targets, indices, num_boxes, log=True)

src/netspresso_trainer/losses/detection/rtdetr.py:176

Methodloss_labels_vfl

(self, outputs, targets, indices, num_boxes, log=True)

src/netspresso_trainer/losses/detection/rtdetr.py:192

Methodmetric_names

(self)

src/netspresso_trainer/metrics/base.py:69

Functionmixnet

(task, conf_model_backbone)

src/netspresso_trainer/models/backbones/experimental/mixnet.py:310

Functionmixtransformer

(task, conf_model_backbone)

src/netspresso_trainer/models/backbones/experimental/mixtransformer.py:237

Functionmobilenetv3

(task, conf_model_backbone)

src/netspresso_trainer/models/backbones/experimental/mobilenetv3.py:162

Functionmobilenetv4

(task, conf_model_backbone)

src/netspresso_trainer/models/backbones/experimental/mobilenetv4.py:202

Functionmobilevit

(task, conf_model_backbone)

src/netspresso_trainer/models/backbones/experimental/mobilevit.py:422

Methodmode

(self)

src/netspresso_trainer/dataloaders/base.py:76

Methodmode

(self)

src/netspresso_trainer/dataloaders/base.py:126

Functionnms

(prediction, nms_thresh=0.45)

tools/device_runtime/postprocessors/detection_postprocessor.py:105

Functionnms

(prediction, nms_thresh=0.45, class_agnostic=False)

src/netspresso_trainer/postprocessors/detection.py:251

Methodnorm_cdf

(x)

src/netspresso_trainer/dataloaders/augmentation/custom/image_proc.py:865

Methodnum_classes

(self)

src/netspresso_trainer/dataloaders/classification.py:213

Methodnum_classes

(self)

src/netspresso_trainer/dataloaders/base.py:60

Methodnum_classes

(self)

src/netspresso_trainer/dataloaders/base.py:114

Methodnum_classes

(self)

src/netspresso_trainer/dataloaders/segmentation.py:265

Functionpad_img

(img, size, fill)

tools/device_runtime/preprocessors/preprocessor.py:41

Functionpidnet

(num_classes: int, conf_model_full)

src/netspresso_trainer/models/full/experimental/pidnet.py:219

Methodprimary_metric

(self)

src/netspresso_trainer/metrics/base.py:73

Methodprocess_annotation

(sample)

src/netspresso_trainer/dataloaders/detection.py:234

Methodprofile_one_epoch

(self)

src/netspresso_trainer/pipelines/train.py:407

Functionrank_filter

(record)

src/netspresso_trainer/utils/logger.py:40

← previous next →901–1,000 of 1,065, ranked by callers