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

Functionread_json

(json_path)

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

Functionreduce_label

(label: np.ndarray)

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

Methodreset_values

(self)

src/netspresso_trainer/losses/builder.py:76

Functionresize_img

(img, size, interpolation, max_size, resize_criteria)

tools/device_runtime/preprocessors/preprocessor.py:30

Functionresnet

ResNet model from "Deep Residual Learning for Image Recognition" https://arxiv.org/pdf/1512.03385.pdf.

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

Methodresult

(self, phase='train')

src/netspresso_trainer/losses/builder.py:72

Methodresult_dir

(self)

src/netspresso_trainer/loggers/base.py:89

Methodroot

(self)

src/netspresso_trainer/dataloaders/base.py:72

Methodroot

(self)

src/netspresso_trainer/dataloaders/base.py:122

Functionrtdetr_decode

(pred, original_shape, num_top_queries=300, score_thresh=0.0)

src/netspresso_trainer/postprocessors/detection.py:31

Functionrtdetr_head

(num_classes, intermediate_features_dim, conf_model_head, **kwargs)

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

Functionrtdetr_hybrid_encoder

(intermediate_features_dim, conf_model_neck, **kwargs)

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

Functionrtmcc

(num_classes, intermediate_features_dim, conf_model_head, **kwargs)

src/netspresso_trainer/models/heads/pose_estimation/experimental/rtmcc.py:400

Methodsample_input

(self)

src/netspresso_trainer/pipelines/base.py:50

Methodsave_dtype

(self)

src/netspresso_trainer/models/base.py:63

Methodsave_summary

(self)

src/netspresso_trainer/pipelines/base.py:76

Methodselect_with_headers

(self, headers: List[str])

demo/func/experiments.py:140

Functionshufflenetv2

(task, conf_model_backbone)

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

Methodstats

(self)

src/netspresso_trainer/dataloaders/base.py:64

Methodstats

(self)

src/netspresso_trainer/dataloaders/base.py:130

Methodstep

Step could be called after every batch update Example: >>> # xdoctest: +SKIP("Undefined vars") >>> scheduler = Cosine

src/netspresso_trainer/schedulers/cosine_warm_restart.py:130

Methodstep_update

(self, num_updates: int)

src/netspresso_trainer/schedulers/base.py:88

Functionstr2bool

(v)

src/netspresso_trainer/utils/engine_utils.py:43

Functionsummary_transform

(phase, task, model_name, yaml_str)

demo/func/augmentation.py:8

Functiontab_model

(args, task_choices, model_choices)

demo/tab/home/model.py:3

Methodtask_support

(self, task)

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

Methodtask_support

(self, task)

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

Methodtask_support

(self, task)

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

Methodtask_support

(self, task)

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

Methodtask_support

(self, task)

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

Methodtask_support

(self, task)

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

Methodtask_support

(self, task)

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

Methodtest_step

(self, test_model, batch)

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

Methodtest_step

(self, test_model, batch)

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

Methodtest_step

(self, test_model, batch)

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

Methodtest_step

(self, test_model, batch)

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

Functiontorch_distributed_barrier

(local_rank: int, distributed: bool)

src/netspresso_trainer/utils/environment.py:70

Methodtrain_loss

(self)

src/netspresso_trainer/pipelines/train.py:150

Methodtrain_step

(self, train_model, batch, optimizer, loss_factory, metric_factory)

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

Methodtrain_step

(self, train_model, batch, optimizer, loss_factory, metric_factory)

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

Methodtrain_step

(self, train_model, batch, optimizer, loss_factory, metric_factory)

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

Methodtrain_step

(self, train_model, batch, optimizer, loss_factory, metric_factory)

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

Functiontrain_transforms_pidnet

(conf_augmentation, training)

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

Functiontrain_with_inline_yaml

()

train.py:20

Functiontransforms_custom

(conf_augmentation, training)

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

Functionunpickle

(file)

tools/open_dataset_tool/wflw.py:136

Functionunpickle

(file)

tools/open_dataset_tool/voc2012_seg.py:59

Methodupdate

(self, val: Union[float, int], n: int = 1)

src/netspresso_trainer/utils/record.py:38

Methodupdate

(self, model)

src/netspresso_trainer/utils/model_ema.py:34

Methodupdate

(self, model)

src/netspresso_trainer/utils/model_ema.py:57

Methodupdate

(self, pred: torch.Tensor, target: torch.Tensor, phase: str, **kwargs: Any)

src/netspresso_trainer/metrics/base.py:48

Methodupdate_epoch

(self, epoch: int)

src/netspresso_trainer/loggers/base.py:92

Methodvalid_loss

(self)

src/netspresso_trainer/pipelines/train.py:154

Methodvalid_loss

(self)

src/netspresso_trainer/pipelines/evaluation.py:69

Methodvalid_step

(self, eval_model, batch, loss_factory, metric_factory)

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

Methodvalid_step

(self, eval_model, batch, loss_factory, metric_factory)

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

Methodvalid_step

(self, eval_model, batch, loss_factory, metric_factory)

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

Methodvalid_step

(self, eval_model, batch, loss_factory, metric_factory)

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

Functionvit

(task, conf_model_backbone)

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

Functionyolo_detection_head

(num_classes, intermediate_features_dim, conf_model_head, **kwargs)

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

Functionyolo_fastest_head_decode

(pred, original_shape, score_thresh=0.7, anchors=None)

src/netspresso_trainer/postprocessors/detection.py:161

Functionyolo_fastest_head_v2

(num_classes, intermediate_features_dim, conf_model_head)

src/netspresso_trainer/models/heads/detection/experimental/yolo_fastest_head.py:53

Functionyolo_head_decode

(pred, original_shape, score_thresh=0.7, anc2vec=None, reg_max=16)

src/netspresso_trainer/postprocessors/detection.py:210

Functionyolopafpn

(intermediate_features_dim, conf_model_neck, **kwargs)

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

Functionyolov9fpn

(intermediate_features_dim, conf_model_neck, **kwargs)

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

← previous1,001–1,065 of 1,065, ranked by callers