Index

_

__getitem__() (torchok.data.datasets.classification.classification.ImageClassificationDataset method) (torchok.data.datasets.detection.detection.DetectionDataset method) (torchok.data.datasets.examples.coco_detection.COCODetection method) (torchok.data.datasets.examples.sop.SOP method) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP method) (torchok.data.datasets.representation.unsupervised_contrastive_dataset.UnsupervisedContrastiveDataset method) (torchok.data.datasets.representation.validation.RetrievalDataset method) (torchok.data.datasets.segmentation.image_segmentation.ImageSegmentationDataset method) __init__() (torchok.callbacks.checkpoint_onnx.CheckpointONNX method) (torchok.callbacks.freeze_unfreeze.FreezeUnfreeze method) (torchok.constructor.logger.MLFlowLoggerX method) (torchok.data.datasets.classification.classification.ImageClassificationDataset method) (torchok.data.datasets.detection.detection.DetectionDataset method) (torchok.data.datasets.examples.coco_detection.COCODetection method) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation method) (torchok.data.datasets.examples.sop.SOP method) (torchok.data.datasets.examples.sweet_pepper.SweetPepper method) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP method) (torchok.data.datasets.representation.unsupervised_contrastive_dataset.UnsupervisedContrastiveDataset method) (torchok.data.datasets.representation.validation.RetrievalDataset method) (torchok.data.datasets.segmentation.image_segmentation.ImageSegmentationDataset method) (torchok.losses.classification.binary_cross_entropy.BCEWithLogitsLoss method) (torchok.losses.segmentation.dice.DiceLoss method) (torchok.metrics.metrics_manager.MetricsManager method) (torchok.metrics.metrics_manager.MetricWithUtils method) (torchok.metrics.representation_ranx.HitAtKMeter method) (torchok.metrics.representation_ranx.MeanAveragePrecisionAtKMeter method) (torchok.metrics.representation_ranx.NDCGAtKMeter method) (torchok.metrics.representation_ranx.PrecisionAtKMeter method) (torchok.metrics.representation_ranx.RecallAtKMeter method) (torchok.metrics.representation_torchmetrics.RetrievalFallOutMeter method) (torchok.metrics.representation_torchmetrics.RetrievalHitRateMeter method) (torchok.metrics.representation_torchmetrics.RetrievalMAPMeter method) (torchok.metrics.representation_torchmetrics.RetrievalMRRMeter method) (torchok.metrics.representation_torchmetrics.RetrievalNormalizedDCGMeter method) (torchok.metrics.representation_torchmetrics.RetrievalPrecisionMeter method) (torchok.metrics.representation_torchmetrics.RetrievalPrecisionRecallCurveMeter method) (torchok.metrics.representation_torchmetrics.RetrievalRecallMeter method) (torchok.metrics.representation_torchmetrics.RetrievalRPrecisionMeter method) (torchok.models.backbones.base_backbone.BackboneWrapper method) (torchok.models.base.BaseModel method) (torchok.models.heads.classification.arcface_head.ArcFaceHead method) (torchok.models.heads.classification.classification_head.ClassificationHead method) (torchok.models.heads.representation.linear_head.LinearHead method) (torchok.models.heads.segmentation.base.SegmentationHead method) (torchok.models.heads.segmentation.ocr.OCRSegmentationHead method) (torchok.models.necks.classification.hrnet.HRNetClassificationNeck method) (torchok.models.necks.segmentation.hrnet.HRNetSegmentationNeck method) (torchok.models.necks.segmentation.unet.UnetNeck method) (torchok.models.poolings.classification.linear.PoolingLinear method) (torchok.models.poolings.classification.pooling.Pooling method) (torchok.tasks.base.BaseTask method) (torchok.tasks.classification.ClassificationTask method) (torchok.tasks.detection.SingleStageDetectionTask method) (torchok.tasks.onnx.ONNXTask method) (torchok.tasks.pairwise_task.PairwiseLearnTask method) (torchok.tasks.segmentation.SegmentationTask method) (torchok.tasks.triplet_task.TripletLearnTask method)

A

annotations_filename (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) annotations_hash (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) annotations_url (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute)

ArcFaceHead (class in torchok.models.heads.classification.arcface_head) as_module() (torchok.tasks.base.BaseTask method) (torchok.tasks.classification.ClassificationTask method) (torchok.tasks.detection.SingleStageDetectionTask method) (torchok.tasks.onnx.ONNXTask method) (torchok.tasks.segmentation.SegmentationTask method)

B

BackboneWrapper (class in torchok.models.backbones.base_backbone) base_folder (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) (torchok.data.datasets.examples.sop.SOP attribute) (torchok.data.datasets.examples.sweet_pepper.SweetPepper attribute) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP attribute) BaseBackbone (class in torchok.models.backbones.base_backbone) BaseModel (class in torchok.models.base)

BaseTask (class in torchok.tasks.base) BCEWithLogitsLoss (class in torchok.losses.classification.binary_cross_entropy) beit_base_patch16_224() (in module torchok.models.backbones.beit) beit_base_patch16_224_in22k() (in module torchok.models.backbones.beit) beit_base_patch16_384() (in module torchok.models.backbones.beit) beit_large_patch16_224() (in module torchok.models.backbones.beit) beit_large_patch16_224_in22k() (in module torchok.models.backbones.beit) beit_large_patch16_384() (in module torchok.models.backbones.beit) beit_large_patch16_512() (in module torchok.models.backbones.beit)

C

calc_loss() (torchok.losses.representation.pairwise.ContrastiveLoss method) calc_relevance_matrix() (torchok.tasks.pairwise_task.PairwiseLearnTask method) CheckpointONNX (class in torchok.callbacks.checkpoint_onnx) CLASSES (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) ClassificationHead (class in torchok.models.heads.classification.classification_head) ClassificationTask (class in torchok.tasks.classification) COCODetection (class in torchok.data.datasets.examples.coco_detection)

COCOSegmentation (class in torchok.data.datasets.examples.coco_segmentation) collate_fn() (torchok.data.datasets.detection.detection.DetectionDataset method) compute() (torchok.metrics.metrics_manager.MetricWithUtils method) configure_optimizers() (torchok.tasks.base.BaseTask method) ContrastiveLoss (class in torchok.losses.representation.pairwise) create_annotation() (torchok.data.datasets.examples.coco_detection.COCODetection method) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation method) create_hooks() (torchok.models.backbones.base_backbone.BaseBackbone method)

D

davit_b() (in module torchok.models.backbones.davit) davit_s() (in module torchok.models.backbones.davit)

davit_t() (in module torchok.models.backbones.davit) DetectionDataset (class in torchok.data.datasets.detection.detection) DiceLoss (class in torchok.losses.segmentation.dice)

E

ecaresnet101d() (in module torchok.models.backbones.resnet) ecaresnet200d() (in module torchok.models.backbones.resnet) ecaresnet269d() (in module torchok.models.backbones.resnet) ecaresnet26t() (in module torchok.models.backbones.resnet) ecaresnet50d() (in module torchok.models.backbones.resnet) ecaresnet50t() (in module torchok.models.backbones.resnet) ecaresnetlight() (in module torchok.models.backbones.resnet) ecaresnext26t_32x4d() (in module torchok.models.backbones.resnet) ecaresnext50t_32x4d() (in module torchok.models.backbones.resnet) efficientnet_b0() (in module torchok.models.backbones.efficientnet) efficientnet_b0_g16_evos() (in module torchok.models.backbones.efficientnet) efficientnet_b0_g8_gn() (in module torchok.models.backbones.efficientnet) efficientnet_b0_gn() (in module torchok.models.backbones.efficientnet) efficientnet_b1() (in module torchok.models.backbones.efficientnet) efficientnet_b2() (in module torchok.models.backbones.efficientnet) efficientnet_b2a() (in module torchok.models.backbones.efficientnet) efficientnet_b3() (in module torchok.models.backbones.efficientnet) efficientnet_b3_g8_gn() (in module torchok.models.backbones.efficientnet) efficientnet_b3_gn() (in module torchok.models.backbones.efficientnet) efficientnet_b3a() (in module torchok.models.backbones.efficientnet) efficientnet_b4() (in module torchok.models.backbones.efficientnet) efficientnet_b5() (in module torchok.models.backbones.efficientnet)

efficientnet_b6() (in module torchok.models.backbones.efficientnet) efficientnet_b7() (in module torchok.models.backbones.efficientnet) efficientnet_b8() (in module torchok.models.backbones.efficientnet) efficientnet_cc_b0_4e() (in module torchok.models.backbones.efficientnet) efficientnet_cc_b0_8e() (in module torchok.models.backbones.efficientnet) efficientnet_cc_b1_8e() (in module torchok.models.backbones.efficientnet) efficientnet_el() (in module torchok.models.backbones.efficientnet) efficientnet_em() (in module torchok.models.backbones.efficientnet) efficientnet_es() (in module torchok.models.backbones.efficientnet) efficientnet_l2() (in module torchok.models.backbones.efficientnet) efficientnet_lite0() (in module torchok.models.backbones.efficientnet) efficientnet_lite1() (in module torchok.models.backbones.efficientnet) efficientnet_lite2() (in module torchok.models.backbones.efficientnet) efficientnet_lite3() (in module torchok.models.backbones.efficientnet) efficientnet_lite4() (in module torchok.models.backbones.efficientnet) efficientnetv2_l() (in module torchok.models.backbones.efficientnet) efficientnetv2_m() (in module torchok.models.backbones.efficientnet) efficientnetv2_rw_m() (in module torchok.models.backbones.efficientnet) efficientnetv2_rw_s() (in module torchok.models.backbones.efficientnet) efficientnetv2_rw_t() (in module torchok.models.backbones.efficientnet) efficientnetv2_s() (in module torchok.models.backbones.efficientnet) efficientnetv2_xl() (in module torchok.models.backbones.efficientnet)

F

fbnetc_100() (in module torchok.models.backbones.efficientnet) fbnetv3_b() (in module torchok.models.backbones.mobilenetv3) fbnetv3_d() (in module torchok.models.backbones.mobilenetv3) fbnetv3_g() (in module torchok.models.backbones.mobilenetv3) filename (torchok.data.datasets.examples.sop.SOP attribute) (torchok.data.datasets.examples.sweet_pepper.SweetPepper attribute) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP attribute) filter_bboxes() (torchok.data.datasets.detection.detection.DetectionDataset method) finalize() (torchok.constructor.logger.MLFlowLoggerX method) FinalizeLogger (class in torchok.callbacks.finalize_logger) finetune_function() (torchok.callbacks.freeze_unfreeze.FreezeUnfreeze method) forward() (torchok.losses.classification.binary_cross_entropy.BCEWithLogitsLoss method) (torchok.losses.segmentation.dice.DiceLoss method) (torchok.models.backbones.base_backbone.BackboneWrapper method) (torchok.models.base.BaseModel method) (torchok.models.heads.classification.arcface_head.ArcFaceHead method) (torchok.models.heads.classification.classification_head.ClassificationHead method) (torchok.models.heads.representation.linear_head.LinearHead method) (torchok.models.heads.segmentation.base.SegmentationHead method) (torchok.models.heads.segmentation.ocr.OCRSegmentationHead method) (torchok.models.necks.classification.hrnet.HRNetClassificationNeck method) (torchok.models.necks.segmentation.hrnet.HRNetSegmentationNeck method) (torchok.models.necks.segmentation.unet.UnetNeck method) (torchok.models.poolings.classification.linear.PoolingLinear method) (torchok.tasks.base.BaseTask method) (torchok.tasks.classification.ClassificationTask method) (torchok.tasks.detection.SingleStageDetectionTask method) (torchok.tasks.onnx.ONNXTask method) (torchok.tasks.segmentation.SegmentationTask method)

forward_features() (torchok.models.backbones.base_backbone.BaseBackbone method) forward_infer_with_gt() (torchok.tasks.onnx.ONNXTask method) forward_with_gt() (torchok.tasks.base.BaseTask method) (torchok.tasks.classification.ClassificationTask method) (torchok.tasks.detection.SingleStageDetectionTask method) (torchok.tasks.onnx.ONNXTask method) (torchok.tasks.pairwise_task.PairwiseLearnTask method) (torchok.tasks.segmentation.SegmentationTask method) (torchok.tasks.triplet_task.TripletLearnTask method) foward_infer() (torchok.tasks.onnx.ONNXTask method) freeze() (torchok.callbacks.freeze_unfreeze.FreezeUnfreeze static method) freeze_before_training() (torchok.callbacks.freeze_unfreeze.FreezeUnfreeze method) FreezeUnfreeze (class in torchok.callbacks.freeze_unfreeze)

G

gc_efficientnetv2_rw_t() (in module torchok.models.backbones.efficientnet) get_raw() (torchok.data.datasets.classification.classification.ImageClassificationDataset method) (torchok.data.datasets.detection.detection.DetectionDataset method) (torchok.data.datasets.examples.sop.SOP method) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP method) (torchok.data.datasets.representation.unsupervised_contrastive_dataset.UnsupervisedContrastiveDataset method) (torchok.data.datasets.representation.validation.RetrievalDataset method) (torchok.data.datasets.segmentation.image_segmentation.ImageSegmentationDataset method)

get_stages() (torchok.models.backbones.base_backbone.BaseBackbone method)

H

HitAtKMeter (class in torchok.metrics.representation_ranx) hrnet_w18() (in module torchok.models.backbones.hrnet) hrnet_w18_small() (in module torchok.models.backbones.hrnet) hrnet_w18_small_v2() (in module torchok.models.backbones.hrnet) hrnet_w30() (in module torchok.models.backbones.hrnet) hrnet_w32() (in module torchok.models.backbones.hrnet)

hrnet_w40() (in module torchok.models.backbones.hrnet) hrnet_w44() (in module torchok.models.backbones.hrnet) hrnet_w48() (in module torchok.models.backbones.hrnet) hrnet_w64() (in module torchok.models.backbones.hrnet) HRNetClassificationNeck (class in torchok.models.necks.classification.hrnet) HRNetSegmentationNeck (class in torchok.models.necks.segmentation.hrnet)

I

ig_resnext101_32x16d() (in module torchok.models.backbones.resnet) ig_resnext101_32x32d() (in module torchok.models.backbones.resnet) ig_resnext101_32x48d() (in module torchok.models.backbones.resnet) ig_resnext101_32x8d() (in module torchok.models.backbones.resnet) ImageClassificationDataset (class in torchok.data.datasets.classification.classification)

ImageSegmentationDataset (class in torchok.data.datasets.segmentation.image_segmentation) in_channels (torchok.models.base.BaseModel property) init_weights() (torchok.models.base.BaseModel method) (torchok.models.poolings.classification.linear.PoolingLinear method) is_number() (torchok.metrics.metrics_manager.MetricsManager static method)

L

label_mapping (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) lcnet_035() (in module torchok.models.backbones.mobilenetv3) lcnet_050() (in module torchok.models.backbones.mobilenetv3)

lcnet_075() (in module torchok.models.backbones.mobilenetv3) lcnet_100() (in module torchok.models.backbones.mobilenetv3) lcnet_150() (in module torchok.models.backbones.mobilenetv3) LinearHead (class in torchok.models.heads.representation.linear_head) log_hyperparams() (torchok.constructor.logger.MLFlowLoggerX method)

M

make_trainable() (torchok.callbacks.freeze_unfreeze.FreezeUnfreeze static method) map_arguments() (torchok.metrics.metrics_manager.MetricWithUtils method) MeanAveragePrecisionAtKMeter (class in torchok.metrics.representation_ranx) MetricsManager (class in torchok.metrics.metrics_manager) MetricWithUtils (class in torchok.metrics.metrics_manager) mixnet_l() (in module torchok.models.backbones.efficientnet) mixnet_m() (in module torchok.models.backbones.efficientnet) mixnet_s() (in module torchok.models.backbones.efficientnet) mixnet_xl() (in module torchok.models.backbones.efficientnet) mixnet_xxl() (in module torchok.models.backbones.efficientnet) MLFlowLoggerX (class in torchok.constructor.logger) mnasnet_050() (in module torchok.models.backbones.efficientnet) mnasnet_075() (in module torchok.models.backbones.efficientnet) mnasnet_100() (in module torchok.models.backbones.efficientnet) mnasnet_140() (in module torchok.models.backbones.efficientnet) mnasnet_a1() (in module torchok.models.backbones.efficientnet) mnasnet_b1() (in module torchok.models.backbones.efficientnet) mnasnet_small() (in module torchok.models.backbones.efficientnet) mobilenetv2_035() (in module torchok.models.backbones.efficientnet) mobilenetv2_050() (in module torchok.models.backbones.efficientnet) mobilenetv2_075() (in module torchok.models.backbones.efficientnet) mobilenetv2_100() (in module torchok.models.backbones.efficientnet) mobilenetv2_110d() (in module torchok.models.backbones.efficientnet) mobilenetv2_120d() (in module torchok.models.backbones.efficientnet) mobilenetv2_140() (in module torchok.models.backbones.efficientnet) mobilenetv3_large_075() (in module torchok.models.backbones.mobilenetv3) mobilenetv3_large_100() (in module torchok.models.backbones.mobilenetv3) mobilenetv3_large_100_miil() (in module torchok.models.backbones.mobilenetv3) mobilenetv3_large_100_miil_in21k() (in module torchok.models.backbones.mobilenetv3) mobilenetv3_rw() (in module torchok.models.backbones.mobilenetv3) mobilenetv3_small_050() (in module torchok.models.backbones.mobilenetv3) mobilenetv3_small_075() (in module torchok.models.backbones.mobilenetv3) mobilenetv3_small_100() (in module torchok.models.backbones.mobilenetv3) module torchok.callbacks.checkpoint_onnx torchok.callbacks.finalize_logger torchok.callbacks.freeze_unfreeze torchok.data.datasets torchok.data.datasets.classification.classification torchok.data.datasets.detection.detection torchok.data.datasets.examples.coco_detection torchok.data.datasets.examples.coco_segmentation torchok.data.datasets.examples.sop torchok.data.datasets.examples.sweet_pepper torchok.data.datasets.examples.triplet_sop torchok.data.datasets.representation.unsupervised_contrastive_dataset torchok.data.datasets.representation.validation torchok.data.datasets.segmentation.image_segmentation torchok.losses.classification.binary_cross_entropy torchok.losses.representation.pairwise torchok.losses.segmentation.dice torchok.metrics.representation_ranx torchok.metrics.representation_torchmetrics torchok.models.backbones torchok.models.backbones.base_backbone torchok.models.backbones.beit torchok.models.backbones.davit torchok.models.backbones.efficientnet torchok.models.backbones.hrnet torchok.models.backbones.mobilenetv3 torchok.models.backbones.resnet torchok.models.backbones.swin torchok.models.base torchok.models.heads.classification.arcface_head torchok.models.heads.classification.classification_head torchok.models.heads.representation.linear_head torchok.models.heads.segmentation.base torchok.models.heads.segmentation.ocr torchok.models.necks.classification.hrnet torchok.models.necks.segmentation.hrnet torchok.models.necks.segmentation.unet torchok.models.poolings.classification.linear torchok.models.poolings.classification.pooling

N

NDCGAtKMeter (class in torchok.metrics.representation_ranx)

no_weight_decay() (torchok.models.base.BaseModel method)

O

OCRSegmentationHead (class in torchok.models.heads.segmentation.ocr) on_epoch_end() (torchok.metrics.metrics_manager.MetricsManager method) on_exception() (torchok.callbacks.finalize_logger.FinalizeLogger method) on_fit_start() (torchok.tasks.base.BaseTask method) on_predict_start() (torchok.tasks.base.BaseTask method) on_test_epoch_end() (torchok.tasks.base.BaseTask method) on_test_start() (torchok.tasks.base.BaseTask method) on_train_batch_end() (torchok.tasks.base.BaseTask method)

on_train_epoch_end() (torchok.tasks.base.BaseTask method) on_validation_batch_end() (torchok.tasks.base.BaseTask method) on_validation_epoch_end() (torchok.tasks.base.BaseTask method) ONNX_EXTENSION (torchok.callbacks.checkpoint_onnx.CheckpointONNX attribute) ONNXTask (class in torchok.tasks.onnx) out_channels (torchok.models.base.BaseModel property) out_encoder_channels (torchok.models.backbones.base_backbone.BackboneWrapper property) (torchok.models.backbones.base_backbone.BaseBackbone property)

P

PairwiseLearnTask (class in torchok.tasks.pairwise_task) Pooling (class in torchok.models.poolings.classification.pooling) PoolingLinear (class in torchok.models.poolings.classification.linear) PrecisionAtKMeter (class in torchok.metrics.representation_ranx)

predict_dataloader() (torchok.tasks.base.BaseTask method) predict_step() (torchok.tasks.base.BaseTask method) (torchok.tasks.detection.SingleStageDetectionTask method) (torchok.tasks.onnx.ONNXTask method) process_function() (torchok.data.datasets.classification.classification.ImageClassificationDataset method)

R

RecallAtKMeter (class in torchok.metrics.representation_ranx) reduction (torchok.losses.classification.binary_cross_entropy.BCEWithLogitsLoss attribute) reset() (torchok.metrics.metrics_manager.MetricWithUtils method) resnet101() (in module torchok.models.backbones.resnet) resnet101d() (in module torchok.models.backbones.resnet) resnet10t() (in module torchok.models.backbones.resnet) resnet14t() (in module torchok.models.backbones.resnet) resnet152() (in module torchok.models.backbones.resnet) resnet152d() (in module torchok.models.backbones.resnet) resnet18() (in module torchok.models.backbones.resnet) resnet18d() (in module torchok.models.backbones.resnet) resnet200() (in module torchok.models.backbones.resnet) resnet200d() (in module torchok.models.backbones.resnet) resnet26() (in module torchok.models.backbones.resnet) resnet26d() (in module torchok.models.backbones.resnet) resnet26t() (in module torchok.models.backbones.resnet) resnet34() (in module torchok.models.backbones.resnet) resnet34d() (in module torchok.models.backbones.resnet) resnet50() (in module torchok.models.backbones.resnet) resnet50_gn() (in module torchok.models.backbones.resnet) resnet50d() (in module torchok.models.backbones.resnet) resnet50t() (in module torchok.models.backbones.resnet) resnetaa101d() (in module torchok.models.backbones.resnet) resnetaa50() (in module torchok.models.backbones.resnet) resnetaa50d() (in module torchok.models.backbones.resnet)

resnetblur101d() (in module torchok.models.backbones.resnet) resnetblur18() (in module torchok.models.backbones.resnet) resnetblur50() (in module torchok.models.backbones.resnet) resnetblur50d() (in module torchok.models.backbones.resnet) resnetrs101() (in module torchok.models.backbones.resnet) resnetrs152() (in module torchok.models.backbones.resnet) resnetrs200() (in module torchok.models.backbones.resnet) resnetrs270() (in module torchok.models.backbones.resnet) resnetrs350() (in module torchok.models.backbones.resnet) resnetrs420() (in module torchok.models.backbones.resnet) resnetrs50() (in module torchok.models.backbones.resnet) resnext101_32x4d() (in module torchok.models.backbones.resnet) resnext101_32x8d() (in module torchok.models.backbones.resnet) resnext101_64x4d() (in module torchok.models.backbones.resnet) resnext50_32x4d() (in module torchok.models.backbones.resnet) resnext50d_32x4d() (in module torchok.models.backbones.resnet) RetrievalDataset (class in torchok.data.datasets.representation.validation) RetrievalFallOutMeter (class in torchok.metrics.representation_torchmetrics) RetrievalHitRateMeter (class in torchok.metrics.representation_torchmetrics) RetrievalMAPMeter (class in torchok.metrics.representation_torchmetrics) RetrievalMRRMeter (class in torchok.metrics.representation_torchmetrics) RetrievalNormalizedDCGMeter (class in torchok.metrics.representation_torchmetrics) RetrievalPrecisionMeter (class in torchok.metrics.representation_torchmetrics) RetrievalPrecisionRecallCurveMeter (class in torchok.metrics.representation_torchmetrics) RetrievalRecallMeter (class in torchok.metrics.representation_torchmetrics) RetrievalRPrecisionMeter (class in torchok.metrics.representation_torchmetrics)

S

SegmentationHead (class in torchok.models.heads.segmentation.base) SegmentationTask (class in torchok.tasks.segmentation) semnasnet_050() (in module torchok.models.backbones.efficientnet) semnasnet_075() (in module torchok.models.backbones.efficientnet) semnasnet_100() (in module torchok.models.backbones.efficientnet) semnasnet_140() (in module torchok.models.backbones.efficientnet) senet154() (in module torchok.models.backbones.resnet) seresnet101() (in module torchok.models.backbones.resnet) seresnet152() (in module torchok.models.backbones.resnet) seresnet152d() (in module torchok.models.backbones.resnet) seresnet18() (in module torchok.models.backbones.resnet) seresnet200d() (in module torchok.models.backbones.resnet) seresnet269d() (in module torchok.models.backbones.resnet) seresnet34() (in module torchok.models.backbones.resnet) seresnet50() (in module torchok.models.backbones.resnet) seresnet50t() (in module torchok.models.backbones.resnet) seresnetaa50d() (in module torchok.models.backbones.resnet) seresnext101_32x4d() (in module torchok.models.backbones.resnet) seresnext101_32x8d() (in module torchok.models.backbones.resnet) seresnext101d_32x8d() (in module torchok.models.backbones.resnet) seresnext26d_32x4d() (in module torchok.models.backbones.resnet) seresnext26t_32x4d() (in module torchok.models.backbones.resnet) seresnext26tn_32x4d() (in module torchok.models.backbones.resnet) seresnext50_32x4d() (in module torchok.models.backbones.resnet) seresnextaa101d_32x8d() (in module torchok.models.backbones.resnet) SingleStageDetectionTask (class in torchok.tasks.detection) SOP (class in torchok.data.datasets.examples.sop)

spnasnet_100() (in module torchok.models.backbones.efficientnet) ssl_resnet18() (in module torchok.models.backbones.resnet) ssl_resnet50() (in module torchok.models.backbones.resnet) ssl_resnext101_32x16d() (in module torchok.models.backbones.resnet) ssl_resnext101_32x4d() (in module torchok.models.backbones.resnet) ssl_resnext101_32x8d() (in module torchok.models.backbones.resnet) ssl_resnext50_32x4d() (in module torchok.models.backbones.resnet) str_type2numpy_type (torchok.tasks.onnx.ONNXTask attribute) SweetPepper (class in torchok.data.datasets.examples.sweet_pepper) swinv2_base_window12_192_22k() (in module torchok.models.backbones.swin) swinv2_base_window12to16_192to256_22kft1k() (in module torchok.models.backbones.swin) swinv2_base_window12to24_192to384_22kft1k() (in module torchok.models.backbones.swin) swinv2_base_window16_256() (in module torchok.models.backbones.swin) swinv2_base_window8_256() (in module torchok.models.backbones.swin) swinv2_custom() (in module torchok.models.backbones.swin) swinv2_large_window12_192_22k() (in module torchok.models.backbones.swin) swinv2_large_window12to16_192to256_22kft1k() (in module torchok.models.backbones.swin) swinv2_large_window12to24_192to384_22kft1k() (in module torchok.models.backbones.swin) swinv2_small_window16_256() (in module torchok.models.backbones.swin) swinv2_small_window8_256() (in module torchok.models.backbones.swin) swinv2_tiny_window16_256() (in module torchok.models.backbones.swin) swinv2_tiny_window8_256() (in module torchok.models.backbones.swin) swsl_resnet18() (in module torchok.models.backbones.resnet) swsl_resnet50() (in module torchok.models.backbones.resnet) swsl_resnext101_32x16d() (in module torchok.models.backbones.resnet) swsl_resnext101_32x4d() (in module torchok.models.backbones.resnet) swsl_resnext101_32x8d() (in module torchok.models.backbones.resnet) swsl_resnext50_32x4d() (in module torchok.models.backbones.resnet)

T

test_csv (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP attribute) test_dataloader() (torchok.tasks.base.BaseTask method) test_step() (torchok.tasks.base.BaseTask method) (torchok.tasks.detection.SingleStageDetectionTask method) (torchok.tasks.onnx.ONNXTask method) test_txt (torchok.data.datasets.examples.sop.SOP attribute) tf_efficientnet_b0() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b0_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b0_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b1() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b1_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b1_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b2() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b2_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b2_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b3() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b3_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b3_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b4() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b4_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b4_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b5() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b5_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b5_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b6() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b6_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b6_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b7() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b7_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b7_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b8() (in module torchok.models.backbones.efficientnet) tf_efficientnet_b8_ap() (in module torchok.models.backbones.efficientnet) tf_efficientnet_cc_b0_4e() (in module torchok.models.backbones.efficientnet) tf_efficientnet_cc_b0_8e() (in module torchok.models.backbones.efficientnet) tf_efficientnet_cc_b1_8e() (in module torchok.models.backbones.efficientnet) tf_efficientnet_el() (in module torchok.models.backbones.efficientnet) tf_efficientnet_em() (in module torchok.models.backbones.efficientnet) tf_efficientnet_es() (in module torchok.models.backbones.efficientnet) tf_efficientnet_l2_ns() (in module torchok.models.backbones.efficientnet) tf_efficientnet_l2_ns_475() (in module torchok.models.backbones.efficientnet) tf_efficientnet_lite0() (in module torchok.models.backbones.efficientnet) tf_efficientnet_lite1() (in module torchok.models.backbones.efficientnet) tf_efficientnet_lite2() (in module torchok.models.backbones.efficientnet) tf_efficientnet_lite3() (in module torchok.models.backbones.efficientnet) tf_efficientnet_lite4() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_b0() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_b1() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_b2() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_b3() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_l() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_l_in21ft1k() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_l_in21k() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_m() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_m_in21ft1k() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_m_in21k() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_s() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_s_in21ft1k() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_s_in21k() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_xl_in21ft1k() (in module torchok.models.backbones.efficientnet) tf_efficientnetv2_xl_in21k() (in module torchok.models.backbones.efficientnet) tf_mixnet_l() (in module torchok.models.backbones.efficientnet) tf_mixnet_m() (in module torchok.models.backbones.efficientnet) tf_mixnet_s() (in module torchok.models.backbones.efficientnet) tf_mobilenetv3_large_075() (in module torchok.models.backbones.mobilenetv3) tf_mobilenetv3_large_100() (in module torchok.models.backbones.mobilenetv3) tf_mobilenetv3_large_minimal_100() (in module torchok.models.backbones.mobilenetv3) tf_mobilenetv3_small_075() (in module torchok.models.backbones.mobilenetv3) tf_mobilenetv3_small_100() (in module torchok.models.backbones.mobilenetv3) tf_mobilenetv3_small_minimal_100() (in module torchok.models.backbones.mobilenetv3) tgz_md5 (torchok.data.datasets.examples.sop.SOP attribute) (torchok.data.datasets.examples.sweet_pepper.SweetPepper attribute) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP attribute) tinynet_a() (in module torchok.models.backbones.efficientnet) tinynet_b() (in module torchok.models.backbones.efficientnet) tinynet_c() (in module torchok.models.backbones.efficientnet) tinynet_d() (in module torchok.models.backbones.efficientnet) tinynet_e() (in module torchok.models.backbones.efficientnet) torchok.callbacks.checkpoint_onnx module torchok.callbacks.finalize_logger module torchok.callbacks.freeze_unfreeze module torchok.data.datasets module torchok.data.datasets.classification.classification module torchok.data.datasets.detection.detection module torchok.data.datasets.examples.coco_detection module torchok.data.datasets.examples.coco_segmentation module torchok.data.datasets.examples.sop module torchok.data.datasets.examples.sweet_pepper module

torchok.data.datasets.examples.triplet_sop module torchok.data.datasets.representation.unsupervised_contrastive_dataset module torchok.data.datasets.representation.validation module torchok.data.datasets.segmentation.image_segmentation module torchok.losses.classification.binary_cross_entropy module torchok.losses.representation.pairwise module torchok.losses.segmentation.dice module torchok.metrics.representation_ranx module torchok.metrics.representation_torchmetrics module torchok.models.backbones module torchok.models.backbones.base_backbone module torchok.models.backbones.beit module torchok.models.backbones.davit module torchok.models.backbones.efficientnet module torchok.models.backbones.hrnet module torchok.models.backbones.mobilenetv3 module torchok.models.backbones.resnet module torchok.models.backbones.swin module torchok.models.base module torchok.models.heads.classification.arcface_head module torchok.models.heads.classification.classification_head module torchok.models.heads.representation.linear_head module torchok.models.heads.segmentation.base module torchok.models.heads.segmentation.ocr module torchok.models.necks.classification.hrnet module torchok.models.necks.segmentation.hrnet module torchok.models.necks.segmentation.unet module torchok.models.poolings.classification.linear module torchok.models.poolings.classification.pooling module train_csv (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) (torchok.data.datasets.examples.sweet_pepper.SweetPepper attribute) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP attribute) train_data_filename (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) train_data_hash (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) train_data_url (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) train_dataloader() (torchok.tasks.base.BaseTask method) train_pkl (torchok.data.datasets.examples.coco_detection.COCODetection attribute) train_txt (torchok.data.datasets.examples.sop.SOP attribute) training (torchok.losses.representation.pairwise.ContrastiveLoss attribute) (torchok.losses.segmentation.dice.DiceLoss attribute) (torchok.metrics.metrics_manager.MetricsManager attribute) (torchok.metrics.metrics_manager.MetricWithUtils attribute) (torchok.models.backbones.base_backbone.BackboneWrapper attribute) (torchok.models.backbones.base_backbone.BaseBackbone attribute) (torchok.models.base.BaseModel attribute) (torchok.models.heads.classification.arcface_head.ArcFaceHead attribute) (torchok.models.heads.classification.classification_head.ClassificationHead attribute) (torchok.models.heads.representation.linear_head.LinearHead attribute) (torchok.models.heads.segmentation.base.SegmentationHead attribute) (torchok.models.heads.segmentation.ocr.OCRSegmentationHead attribute) (torchok.models.necks.classification.hrnet.HRNetClassificationNeck attribute) (torchok.models.necks.segmentation.hrnet.HRNetSegmentationNeck attribute) (torchok.models.necks.segmentation.unet.UnetNeck attribute) (torchok.models.poolings.classification.linear.PoolingLinear attribute) (torchok.models.poolings.classification.pooling.Pooling attribute) training_step() (torchok.tasks.base.BaseTask method) (torchok.tasks.detection.SingleStageDetectionTask method) TRIPLET_SOP (class in torchok.data.datasets.examples.triplet_sop) TripletLearnTask (class in torchok.tasks.triplet_task) tv_resnet101() (in module torchok.models.backbones.resnet) tv_resnet152() (in module torchok.models.backbones.resnet) tv_resnet34() (in module torchok.models.backbones.resnet) tv_resnet50() (in module torchok.models.backbones.resnet) tv_resnext50_32x4d() (in module torchok.models.backbones.resnet)

U

UnetNeck (class in torchok.models.necks.segmentation.unet) unfreeze_and_add_param_group() (torchok.callbacks.freeze_unfreeze.FreezeUnfreeze static method) UnsupervisedContrastiveDataset (class in torchok.data.datasets.representation.unsupervised_contrastive_dataset) update() (torchok.metrics.metrics_manager.MetricsManager method) (torchok.metrics.metrics_manager.MetricWithUtils method)

url (torchok.data.datasets.examples.sop.SOP attribute) (torchok.data.datasets.examples.sweet_pepper.SweetPepper attribute) (torchok.data.datasets.examples.triplet_sop.TRIPLET_SOP attribute)

V

val_dataloader() (torchok.tasks.base.BaseTask method) valid_csv (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) (torchok.data.datasets.examples.sweet_pepper.SweetPepper attribute) valid_data_filename (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) valid_data_hash (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute)

valid_data_url (torchok.data.datasets.examples.coco_detection.COCODetection attribute) (torchok.data.datasets.examples.coco_segmentation.COCOSegmentation attribute) valid_pkl (torchok.data.datasets.examples.coco_detection.COCODetection attribute) validation_step() (torchok.tasks.base.BaseTask method) (torchok.tasks.detection.SingleStageDetectionTask method) (torchok.tasks.triplet_task.TripletLearnTask method)

W

wide_resnet101_2() (in module torchok.models.backbones.resnet)

wide_resnet50_2() (in module torchok.models.backbones.resnet)