/myDetection

A personal repository of image & video object detection using PyTorch.

Primary LanguagePython

myDetection

  • YOLO v3
  • YOLO v3 + FCOS
  • EfficentNet backbone
  • EfficientNet + BiFPN + YOLO
  • EfficientDet
  • EfficientDet + FCOS
  • EfficientDet + custom FCOS
  • EfficientDet + custom FCOS + ATSS
  • Hard example mining
  • Ultralytics YOLO
  • Attention
  • Mosaic training data augmentation
  • DETR
  • Feature aggregation
  • EfficientDet + FCOS + angle
  • EfficientDet + FCOS + mask

Installation

  1. Clone repository
  2. conda develop --name pytorch_env /home/duanzh/Projects/myDetection

Uninstall

conda develop --uninstall /home/duanzh/Projects/myDetection