cutcutcodec.core.nn.model.enhancement.train

Utils to train and create a dataset.

Functions

plot([log])	Draw the training curves.
train(dataset[, model, log])	Train the model.

Details

cutcutcodec.core.nn.model.enhancement.train.plot(log: Path | str | bytes | None = None)

Draw the training curves.

cutcutcodec.core.nn.model.enhancement.train.train(dataset: VideoDataset | Path | str | bytes, model: CNN | None = None, log: Path | str | bytes | None = None)

Train the model.

Parameters

datasetVideoDataset or pathlike

The dataset containing the videos and dataaug or the folder.

modelCNN, optional

Use the default constructor if not provided.

logpathlike

The directory to store the logs.

Examples

>>> from cutcutcodec.core.nn.model.enhancement.train import train
>>> # train("~/dataset/video_xiph")
>>>