Skip to content

shaunyuan22/Unc-SOD

Repository files navigation

Unc-SOD: An uncertainty Learning Framework for Small Object Detection

📢 Introduction

This is the official repository of our paper titled Unc-SOD: An uncertainty Learning Framework for Small Object Detection, and the full implementation will be available after the peer-review.

🎡 Dependencies

  • CUDA 11.3
  • Python 3.8
  • PyTorch 1.10.0
  • TorchVision 0.11.0
  • mmcv-full 1.5.0
  • numpy 1.22.4

📂 Datasets