src.yolov5 package

Subpackages

• src.yolov5.models package
  • Submodules
  • src.yolov5.models.common module
    • AutoShape
      • AutoShape.autoshape()
      • AutoShape.classes
      • AutoShape.conf
      • AutoShape.forward()
      • AutoShape.iou
      • AutoShape.max_det
      • AutoShape.training
    • Bottleneck
      • Bottleneck.forward()
      • Bottleneck.training
    • BottleneckCSP
      • BottleneckCSP.forward()
      • BottleneckCSP.training
    • C3
      • C3.forward()
      • C3.training
    • C3TR
      • C3TR.training
    • Classify
      • Classify.forward()
      • Classify.training
    • Concat
      • Concat.forward()
      • Concat.training
    • Contract
      • Contract.forward()
      • Contract.training
    • Conv
      • Conv.forward()
      • Conv.fuseforward()
      • Conv.training
    • DWConv()
    • Detections
      • Detections.crop()
      • Detections.display()
      • Detections.pandas()
      • Detections.print()
      • Detections.render()
      • Detections.save()
      • Detections.show()
      • Detections.tolist()
    • Expand
      • Expand.forward()
      • Expand.training
    • Focus
      • Focus.forward()
      • Focus.training
    • NMS
      • NMS.classes
      • NMS.conf
      • NMS.forward()
      • NMS.iou
      • NMS.max_det
      • NMS.training
    • SPP
      • SPP.forward()
      • SPP.training
    • TransformerBlock
      • TransformerBlock.forward()
      • TransformerBlock.training
    • TransformerLayer
      • TransformerLayer.forward()
      • TransformerLayer.training
    • autopad()
  • src.yolov5.models.experimental module
    • CrossConv
      • CrossConv.forward()
      • CrossConv.training
    • Ensemble
      • Ensemble.forward()
      • Ensemble.training
    • GhostBottleneck
      • GhostBottleneck.forward()
      • GhostBottleneck.training
    • GhostConv
      • GhostConv.forward()
      • GhostConv.training
    • MixConv2d
      • MixConv2d.forward()
      • MixConv2d.training
    • Sum
      • Sum.forward()
      • Sum.training
    • attempt_load()
  • src.yolov5.models.export module
  • src.yolov5.models.yolo module
    • Detect
      • Detect.forward()
      • Detect.onnx_dynamic
      • Detect.stride
      • Detect.training
    • Model
      • Model.autoshape()
      • Model.forward()
      • Model.forward_augment()
      • Model.forward_once()
      • Model.fuse()
      • Model.info()
      • Model.nms()
      • Model.training
    • parse_model()
  • Module contents
• src.yolov5.utils package
  • Subpackages
    • src.yolov5.utils.aws package
      • Submodules
      • src.yolov5.utils.aws.resume module
      • Module contents
    • src.yolov5.utils.wandb_logging package
      • Submodules
      • src.yolov5.utils.wandb_logging.log_dataset module
      • src.yolov5.utils.wandb_logging.wandb_utils module
      • Module contents
  • Submodules
  • src.yolov5.utils.activations module
    • AconC
      • AconC.forward()
      • AconC.training
    • FReLU
      • FReLU.forward()
      • FReLU.training
    • Hardswish
      • Hardswish.forward()
      • Hardswish.training
    • MemoryEfficientMish
      • MemoryEfficientMish.F
      • MemoryEfficientMish.forward()
      • MemoryEfficientMish.training
    • MetaAconC
      • MetaAconC.forward()
      • MetaAconC.training
    • Mish
      • Mish.forward()
      • Mish.training
    • SiLU
      • SiLU.forward()
      • SiLU.training
  • src.yolov5.utils.autoanchor module
    • check_anchor_order()
    • check_anchors()
    • kmean_anchors()
  • src.yolov5.utils.datasets module
    • InfiniteDataLoader
      • InfiniteDataLoader.batch_size
      • InfiniteDataLoader.dataset
      • InfiniteDataLoader.drop_last
      • InfiniteDataLoader.num_workers
      • InfiniteDataLoader.pin_memory
      • InfiniteDataLoader.prefetch_factor
      • InfiniteDataLoader.sampler
      • InfiniteDataLoader.timeout
    • LoadImages
      • LoadImages.new_video()
    • LoadImagesAndLabels
      • LoadImagesAndLabels.cache_labels()
      • LoadImagesAndLabels.collate_fn()
      • LoadImagesAndLabels.collate_fn4()
    • LoadStreams
      • LoadStreams.update()
    • LoadWebcam
    • augment_hsv()
    • autosplit()
    • box_candidates()
    • create_dataloader()
    • create_folder()
    • cutout()
    • exif_size()
    • extract_boxes()
    • flatten_recursive()
    • get_hash()
    • hist_equalize()
    • img2label_paths()
    • letterbox()
    • load_image()
    • load_mosaic()
    • load_mosaic9()
    • random_perspective()
    • replicate()
  • src.yolov5.utils.general module
    • apply_classifier()
    • bbox_iou()
    • box_iou()
    • check_dataset()
    • check_file()
    • check_git_status()
    • check_img_size()
    • check_imshow()
    • check_online()
    • check_python()
    • check_requirements()
    • clean_str()
    • clip_coords()
    • coco80_to_coco91_class()
    • colorstr()
    • download()
    • emojis()
    • file_size()
    • get_latest_run()
    • increment_path()
    • init_seeds()
    • is_colab()
    • is_docker()
    • labels_to_class_weights()
    • labels_to_image_weights()
    • make_divisible()
    • non_max_suppression()
    • one_cycle()
    • print_mutation()
    • resample_segments()
    • save_one_box()
    • scale_coords()
    • segment2box()
    • segments2boxes()
    • set_logging()
    • strip_optimizer()
    • wh_iou()
    • xyn2xy()
    • xywh2xyxy()
    • xywhn2xyxy()
    • xyxy2xywh()
  • src.yolov5.utils.google_utils module
    • attempt_download()
    • gdrive_download()
    • get_token()
    • gsutil_getsize()
    • safe_download()
  • src.yolov5.utils.loss module
    • BCEBlurWithLogitsLoss
      • BCEBlurWithLogitsLoss.forward()
      • BCEBlurWithLogitsLoss.training
    • ComputeLoss
      • ComputeLoss.build_targets()
    • FocalLoss
      • FocalLoss.forward()
      • FocalLoss.training
    • QFocalLoss
      • QFocalLoss.forward()
      • QFocalLoss.training
    • smooth_BCE()
  • src.yolov5.utils.metrics module
    • ConfusionMatrix
      • ConfusionMatrix.matrix()
      • ConfusionMatrix.plot()
      • ConfusionMatrix.print()
      • ConfusionMatrix.process_batch()
    • ap_per_class()
    • compute_ap()
    • fitness()
    • plot_mc_curve()
    • plot_pr_curve()
  • src.yolov5.utils.plots module
    • Colors
      • Colors.hex2rgb()
    • butter_lowpass_filtfilt()
    • hist2d()
    • output_to_target()
    • plot_evolution()
    • plot_images()
    • plot_labels()
    • plot_lr_scheduler()
    • plot_one_box()
    • plot_one_box_PIL()
    • plot_results()
    • plot_results_overlay()
    • plot_study_txt()
    • plot_targets_txt()
    • plot_test_txt()
    • plot_wh_methods()
    • profile_idetection()
  • src.yolov5.utils.torch_utils module
    • ModelEMA
      • ModelEMA.update()
      • ModelEMA.update_attr()
    • copy_attr()
    • date_modified()
    • de_parallel()
    • find_modules()
    • fuse_conv_and_bn()
    • git_describe()
    • init_torch_seeds()
    • initialize_weights()
    • intersect_dicts()
    • is_parallel()
    • load_classifier()
    • model_info()
    • profile()
    • prune()
    • scale_img()
    • select_device()
    • sparsity()
    • time_synchronized()
    • torch_distributed_zero_first()
  • Module contents

Module contents