Path	Lines of Code
common/__init__.py	9
config/classification/mobilenetv2.yaml	85
config/classification/mobilevit_small.yaml	90
config/classification/mobilevit_x_small.yaml	90
config/classification/mobilevit_xx_small.yaml	90
config/classification/resnet.yaml	82
config/detection/ssd_mobilevit_small_320.yaml	93
config/detection/ssd_mobilevit_x_small_320.yaml	93
config/detection/ssd_mobilevit_xx_small_320.yaml	93
config/segmentation/deeplabv3_mobilevit_small.yaml	109
config/segmentation/deeplabv3_mobilevit_x_small.yaml	109
config/segmentation/deeplabv3_mobilevit_xx_small.yaml	109
cvnets/__init__.py	6
cvnets/layers/__init__.py	64
cvnets/layers/activation/__init__.py	47
cvnets/layers/activation/gelu.py	8
cvnets/layers/activation/hard_sigmoid.py	8
cvnets/layers/activation/hard_swish.py	8
cvnets/layers/activation/leaky_relu.py	8
cvnets/layers/activation/prelu.py	8
cvnets/layers/activation/relu.py	8
cvnets/layers/activation/relu6.py	8
cvnets/layers/activation/sigmoid.py	8
cvnets/layers/activation/swish.py	8
cvnets/layers/adaptive_pool.py	7
cvnets/layers/base_layer.py	15
cvnets/layers/conv_layer.py	218
cvnets/layers/dropout.py	13
cvnets/layers/flatten.py	7
cvnets/layers/global_pool.py	37
cvnets/layers/identity.py	9
cvnets/layers/linear_layer.py	137
cvnets/layers/multi_head_attention.py	82
cvnets/layers/non_linear_layers.py	29
cvnets/layers/normalization/__init__.py	49
cvnets/layers/normalization/batch_norm.py	32
cvnets/layers/normalization/group_norm.py	20
cvnets/layers/normalization/instance_norm.py	32
cvnets/layers/normalization/layer_norm.py	18
cvnets/layers/normalization/sync_batch_norm.py	17
cvnets/layers/normalization_layers.py	87
cvnets/layers/pixel_shuffle.py	9
cvnets/layers/pooling.py	30
cvnets/layers/positional_encoding.py	23
cvnets/layers/random_layers.py	37
cvnets/layers/sep_conv_layer.py	56
cvnets/layers/upsample.py	8
cvnets/misc/__init__.py	1
cvnets/misc/anchor_generator.py	71
cvnets/misc/averaging_utils.py	35
cvnets/misc/box_utils.py	34
cvnets/misc/common.py	34
cvnets/misc/init_utils.py	75
cvnets/misc/match_prior.py	44
cvnets/misc/profiler.py	15
cvnets/models/__init__.py	27
cvnets/models/classification/__init__.py	91
cvnets/models/classification/base_cls.py	153
cvnets/models/classification/config/__init__.py	1
cvnets/models/classification/config/mobilenetv2.py	47
cvnets/models/classification/config/mobilevit.py	198
cvnets/models/classification/config/resnet.py	111
cvnets/models/classification/mobilenetv2.py	128
cvnets/models/classification/mobilevit.py	177
cvnets/models/classification/resnet.py	101
cvnets/models/detection/__init__.py	81
cvnets/models/detection/base_detection.py	83
cvnets/models/detection/ssd.py	282
cvnets/models/segmentation/__init__.py	111
cvnets/models/segmentation/base_seg.py	32
cvnets/models/segmentation/enc_dec.py	62
cvnets/models/segmentation/heads/__init__.py	44
cvnets/models/segmentation/heads/base_seg_head.py	71
cvnets/models/segmentation/heads/deeplabv3.py	90
cvnets/modules/__init__.py	23
cvnets/modules/aspp_block.py	123
cvnets/modules/base_module.py	12
cvnets/modules/feature_pyramid.py	63
cvnets/modules/mobilenetv2.py	114
cvnets/modules/mobilevit_block.py	165
cvnets/modules/ppm.py	67
cvnets/modules/resnet.py	113
cvnets/modules/squeeze_excitation.py	46
cvnets/modules/ssd.py	74
cvnets/modules/transformer.py	47
data/__init__.py	1
data/data_loaders.py	64
data/datasets/__init__.py	116
data/datasets/classification/__init__.py	1
data/datasets/classification/imagenet.py	62
data/datasets/dataset_base.py	68
data/datasets/detection/__init__.py	1
data/datasets/detection/coco.py	222
data/datasets/segmentation/__init__.py	1
data/datasets/segmentation/pascal_voc.py	123
data/sampler/__init__.py	51
data/sampler/base_sampler.py	68
data/sampler/batch_sampler.py	75
data/sampler/utils.py	34
data/sampler/variable_batch_sampler.py	231
data/transforms/__init__.py	31
data/transforms/base_transforms.py	13
data/transforms/image.py	1005
engine/__init__.py	2
engine/detection_utils/__init__.py	1
engine/detection_utils/coco_map.py	58
engine/eval_detection.py	259
engine/eval_segmentation.py	326
engine/evaluation_engine.py	67
engine/segmentation_utils/__init__.py	1
engine/segmentation_utils/cityscapes_iou.py	26
engine/training_engine.py	309
engine/utils.py	30
loss_fn/__init__.py	50
loss_fn/base_criteria.py	24
loss_fn/classification.py	35
loss_fn/classification_loss_fns/__init__.py	22
loss_fn/classification_loss_fns/cross_entropy.py	31
loss_fn/classification_loss_fns/label_smoothing.py	67
loss_fn/detection.py	37
loss_fn/detection_loss_fns/__init__.py	21
loss_fn/detection_loss_fns/ssd_multibox_loss.py	80
loss_fn/distillation.py	33
loss_fn/distillation_loss_fns/__init__.py	21
loss_fn/distillation_loss_fns/vanilla.py	134
loss_fn/segmentation.py	34
loss_fn/segmentation_loss_fns/__init__.py	21
loss_fn/segmentation_loss_fns/cross_entropy.py	63
main_conversion.py	44
main_eval.py	100
main_train.py	173
metrics/__init__.py	30
metrics/intersection_over_union.py	29
metrics/metric_monitor.py	27
metrics/stats.py	107
metrics/topk_accuracy.py	19
optim/__init__.py	62
optim/adam.py	43
optim/adamw.py	36
optim/base_optim.py	9
optim/scheduler/__init__.py	54
optim/scheduler/base_scheduler.py	29
optim/scheduler/cosine.py	47
optim/scheduler/cyclic.py	121
optim/scheduler/polynomial.py	46
optim/sgd.py	40
options/__init__.py	1
options/opts.py	110
options/utils.py	48
setup.py	62
utils/__init__.py	1
utils/checkpoint_utils.py	174
utils/color_map.py	40
utils/common_utils.py	59
utils/ddp_utils.py	37
utils/download_utils.py	20
utils/logger.py	45
utils/math_utils.py	12
utils/pytorch_to_coreml.py	54
utils/tensor_utils.py	47