Types & classes — github.com/JDSobek/MedYOLO

CodeHub MCPcopy Create free account

hub / github.com/JDSobek/MedYOLO / types & classes

Types & classes18 in github.com/JDSobek/MedYOLO

⨍ Functions 151◇ Types & classes 18

↓ 7 callersClassConv

models3D/model.py:130

↓ 3 callersClassLoadNiftisAndLabels

YOLO3D Pytorch Dataset for training.

utils3D/datasets.py:140

↓ 3 callersClassModel

models3D/model.py:393

↓ 2 callersClassCallbacks

Handles all registered callbacks for YOLOv5 Hooks

utils/callbacks.py:7

↓ 1 callersClassBottleneck

models3D/model.py:155

↓ 1 callersClassComputeLossVF

Compute losses in 3D

utils3D/lossandmetrics.py:192

↓ 1 callersClassConfusionMatrix

utils3D/lossandmetrics.py:369

↓ 1 callersClassEarlyStopping

utils/torch_utils.py:84

↓ 1 callersClassEnsemble

models/experimental.py:10

↓ 1 callersClassFocalLoss

utils/loss.py:15

↓ 1 callersClassLoadNiftis

YOLO3D Pytorch Dataset for inference.

utils3D/datasets.py:91

↓ 1 callersClassModelEMA

Model Exponential Moving Average from https://github.com/rwightman/pytorch-image-models Keep a moving average of everything in the model state_di

utils/torch_utils.py:107

↓ 1 callersClass_RepeatSampler

Sampler that repeats forever Args: sampler (Sampler)

utils/datasets.py:46

models3D/model.py:176

models/common.py:17

models3D/model.py:225

ClassInfiniteDataLoader

Dataloader that reuses workers Uses same syntax as vanilla DataLoader

utils/datasets.py:27

models3D/model.py:201