galaxy.models.yolo_class

Classes

Functions

load_model(→ torch.nn.Module)

Module Contents

class galaxy.models.yolo_class.YOLO_Class(yolo_model: torch.nn.Module)

Bases: torch.nn.Module

model

forward(x)

galaxy.models.yolo_class.load_model(in_channels: int, n_out_classes: int = 1) → torch.nn.Module