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

Method_get_closed_form_lr

(self)

src/netspresso_trainer/schedulers/multi_step_lr.py:64

Method_get_name

(self)

src/netspresso_trainer/models/base.py:70

Method_get_name

(self)

src/netspresso_trainer/models/base.py:150

Method_get_tgt_permutation_idx

(self, indices)

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

Method_init_weights

(self, m)

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:101

Function_is_single_task_model

(conf_model: DictConfig)

demo/func/experiments.py:28

Method_load

(i, samples)

src/netspresso_trainer/dataloaders/classification.py:145

Method_load

(i, samples)

src/netspresso_trainer/dataloaders/detection.py:128

Method_load

(i, samples)

src/netspresso_trainer/dataloaders/pose_estimation.py:125

Method_load

(i, samples)

src/netspresso_trainer/dataloaders/segmentation.py:161

Method_load_dataset

(self, root, subset_name=None, cache_dir=None)

src/netspresso_trainer/dataloaders/base.py:94

Method_to_numpy

(tensor: torch.Tensor)

src/netspresso_trainer/loggers/base.py:98

Methodadd_arguments

(cls, parser: argparse.ArgumentParser)

src/netspresso_trainer/models/op/custom.py:757

Functionall_mlp_decoder

(num_classes, intermediate_features_dim, conf_model_head, **kwargs)

src/netspresso_trainer/models/heads/segmentation/experimental/all_mlp_decoder.py:78

Functionanchor_decoupled_head

(num_classes, intermediate_features_dim, conf_model_head, **kwargs)

src/netspresso_trainer/models/heads/detection/experimental/anchor_decoupled_head.py:166

Functionanchor_decoupled_head_decode

(pred, original_shape, topk_candidates=1000, score_thresh=0.05)

src/netspresso_trainer/postprocessors/detection.py:55

Functionanchor_free_decoupled_head

(num_classes, intermediate_features_dim, conf_model_head, **kwargs)

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

Functionanchor_free_decoupled_head_decode

(pred, original_shape, score_thresh=0.7)

tools/device_runtime/postprocessors/detection_postprocessor.py:22

Functionanchor_free_decoupled_head_decode

(pred, original_shape, score_thresh=0.7)

src/netspresso_trainer/postprocessors/detection.py:114

Methodavg

(self)

src/netspresso_trainer/utils/record.py:49

Methodavg

(self)

src/netspresso_trainer/utils/record.py:78

Methodavg

(self)

src/netspresso_trainer/metrics/segmentation/metric.py:48

Methodcache_dataset

(self)

src/netspresso_trainer/dataloaders/classification.py:142

Methodcache_dataset

(self)

src/netspresso_trainer/dataloaders/classification.py:242

Methodcache_dataset

(self)

src/netspresso_trainer/dataloaders/detection.py:125

Methodcache_dataset

(self)

src/netspresso_trainer/dataloaders/base.py:53

Methodcache_dataset

(self)

src/netspresso_trainer/dataloaders/pose_estimation.py:122

Methodcache_dataset

(self)

src/netspresso_trainer/dataloaders/segmentation.py:158

Methodcache_dataset

(self)

src/netspresso_trainer/dataloaders/segmentation.py:318

Methodcalibrate

(self, pred, target, **kwargs)

src/netspresso_trainer/metrics/pose_estimation/metric.py:74

Methodcalibrate

(self, predictions, targets, **kwargs)

src/netspresso_trainer/metrics/detection/metric.py:310

Methodcalibrate

(self, predictions, targets, **kwargs)

src/netspresso_trainer/metrics/detection/metric.py:332

Methodcalibrate

(self, predictions, targets, **kwargs)

src/netspresso_trainer/metrics/detection/metric.py:354

Methodcalibrate

(self, predictions, targets, **kwargs)

src/netspresso_trainer/metrics/detection/metric.py:375

Methodcalibrate

(self, predictions, targets, **kwargs)

src/netspresso_trainer/metrics/detection/metric.py:395

Methodcalibrate

(self, pred, target, **kwargs)

src/netspresso_trainer/metrics/classification/metric.py:54

Methodcalibrate

(self, pred, target, **kwargs)

src/netspresso_trainer/metrics/classification/metric.py:73

Methodcalibrate

(self, pred, target, **kwargs)

src/netspresso_trainer/metrics/segmentation/metric.py:98

Methodcalibrate

(self, pred, target, **kwargs)

src/netspresso_trainer/metrics/segmentation/metric.py:136

Functionchange_tab_to

(destination=None)

demo/tab/home/train.py:13

Functionchange_tab_to_augmentation

()

demo/tab/home/main.py:35

Functionchange_tab_to_dataset

()

demo/tab/home/main.py:31

Functionchange_tab_to_pynetspresso

()

demo/app.py:20

Functionchange_tab_to_scheduler

()

demo/tab/home/main.py:39

Functionchange_tab_to_train

()

demo/tab/home/main.py:27

Methodclass_map

(self)

src/netspresso_trainer/dataloaders/classification.py:217

Methodclass_map

(self)

src/netspresso_trainer/dataloaders/base.py:68

Methodclass_map

(self)

src/netspresso_trainer/dataloaders/base.py:118

Methodclass_map

(self)

src/netspresso_trainer/dataloaders/segmentation.py:269

Functionclassification_mix_collate_fn

(original_batch, mix_transforms)

src/netspresso_trainer/dataloaders/utils/collate_fn.py:23

Functionclassification_onehot_collate_fn

(original_batch, num_classes)

src/netspresso_trainer/dataloaders/utils/collate_fn.py:42

Functioncompress_with_session

( session: NetsPressoSession, model_name: str, model_path: Union[Path, str], batch_size: int, chan

demo/func/pynetspresso.py:84

Functioncopy_config

(input_config)

demo/tab/home/main.py:23

Functioncreate_transform

(model_name: str, is_training=False)

src/netspresso_trainer/dataloaders/augmentation/transforms.py:93

Functioncspdarknet

(task, conf_model_backbone)

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

Methoddefault

(self)

demo/func/experiments.py:132

Functiondefault_collate_fn

(original_batch)

src/netspresso_trainer/dataloaders/utils/collate_fn.py:61

Methoddefault_no_render

(self)

demo/func/experiments.py:137

Functiondeformable_attention_core_func

Args: value (Tensor): [bs, value_length, n_head, c] value_spatial_shapes (Tensor|List): [n_levels, 2] value_level_start_i

src/netspresso_trainer/models/heads/detection/experimental/rtdetr_head.py:37

Methoddone

(self)

src/netspresso_trainer/utils/record.py:91

Functiondownload_worker

(i, split, base_url, images_dir, silent=True)

tools/open_dataset_tool/objects365.py:102

Functionefficientformer

(task, conf_model_backbone)

src/netspresso_trainer/models/backbones/experimental/efficientformer.py:419

Methodelapsed

(self)

src/netspresso_trainer/utils/record.py:95

Methodeval

Set the model to evaluation mode.

src/netspresso_trainer/models/base.py:256

Functionexpand_to_chs

(x, n)

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

Functionfast_collate

A fast collation function optimized for uint8 images (np array or torch) and int64 targets (labels)

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

Functionfc

(feature_dim, num_classes, conf_model_head, **kwargs)

src/netspresso_trainer/models/heads/classification/core/fc.py:79

Functionfc_conv

(feature_dim, num_classes, conf_model_head, **kwargs)

src/netspresso_trainer/models/heads/classification/core/fc.py:82

Methodfeature_dim

(self)

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

Methodfeature_dim

(self)

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

Methodfeature_dim

(self)

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

Methodfeature_dim

(self)

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

Methodfeature_dim

(self)

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

Methodfeature_dim

(self)

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

Methodfeature_dim

(self)

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

Methodfeature_dim

(self)

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

Methodfilter_with

(self, task, data, model, macs, params, ignore_compressed_model)

demo/func/experiments.py:152

Methodfind_compression_info_with_id

(self, id: str)

demo/func/experiments.py:182

Methodflatten_augmentation_config

(self, aug_config, parent_key="")

src/netspresso_trainer/loggers/mlflow.py:231

Methodforward

(self, image, label=None, mask=None, bbox=None, keypoint=None, dataset=None)

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

Methodforward

(self, image, label=None, mask=None, bbox=None, keypoint=None, dataset=None)

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

Methodforward

(self, image, label=None, mask=None, bbox=None, keypoint=None, dataset=None)

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

Methodforward

(self, image, label=None, mask=None, bbox=None, keypoint=None, dataset=None)

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

Methodforward

(self, image, label=None, mask=None, bbox=None, keypoint=None, dataset=None)

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

Methodforward

(self, image, label=None, mask=None, bbox=None, keypoint=None, dataset=None)

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

Methodforward

img (PIL Image or Tensor): Image to be transformed. Returns: PIL Image or Tensor: AutoAugmented image.

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

Methodforward

(self, x)

src/netspresso_trainer/models/base.py:77

Methodforward

(self, x)

src/netspresso_trainer/models/base.py:92

Methodforward

(self, x)

src/netspresso_trainer/models/base.py:102

Methodforward

(self, x)

src/netspresso_trainer/models/base.py:114

Methodforward

(self, x)

src/netspresso_trainer/models/base.py:126

Methodforward

Args: inputs: List of backbone features [P3, P4, P5] Returns: Tuple[Tensor]: FPN features at different scales

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

Methodforward

Forward function.

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

Methodforward

(self, inputs)

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

Methodforward

Args: inputs: input images. Returns: Tuple[Tensor]: FPN feature.

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

Methodforward

(self, src, src_mask=None, pos_embed=None)

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

Methodforward

(self, src, src_mask=None, pos_embed=None)

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

Methodforward

(self, feats)

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

Methodforward

(self, x: Tensor)

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

Methodforward

(self, x)

src/netspresso_trainer/models/backbones/experimental/vit.py:77

← previous next →701–800 of 1,065, ranked by callers