s6.app.cog.keypoint

Keypoint training, preview, and ONNX export utilities.

s6.app.cog.keypoint.load_dataset_config(config_path: str) → Config

s6.app.cog.keypoint.build_model(config: Config, pretrained_backbone: bool = False) → GenericKeypointModel

s6.app.cog.keypoint.unpack_batch(batch) → Tuple[Tensor, Tensor, Tensor | None]

s6.app.cog.keypoint.unpack_model_outputs(outputs) → Tuple[Tensor, Tensor, Tensor | None]

s6.app.cog.keypoint.log_graph(writer, model, dataloader, device)

s6.app.cog.keypoint.batch_draw_keypoints(images: Tensor, keypoints: Tensor, masks: Tensor | None = None) → Tensor

s6.app.cog.keypoint.train_step(batch, model, optimizer, keypoint_criterion, segmentation_criterion, device, segmentation_loss_weight: float = 1.0, precision: str = 'fp32')

s6.app.cog.keypoint.make_train_dataloader(args)

s6.app.cog.keypoint.load_deploy_output_size(config_path: str) → int

s6.app.cog.keypoint.resolve_precision(args: Namespace) → str

s6.app.cog.keypoint.resolve_deploy_path(args: Namespace, config: Config) → str

s6.app.cog.keypoint.train(args)

s6.app.cog.keypoint.preview_dataset(args)

s6.app.cog.keypoint.deploy_model(args)

s6.app.cog.keypoint.main()