YOLC项目安装与配置指南

1. 项目基础介绍

YOLC（You Only Look Clusters）是一个用于微小物体检测的开源项目，专为航拍图像设计。该项目基于聚类算法，针对微小目标检测提出了一种有效的方法。项目的主要编程语言是Python。

2. 关键技术与框架

• 编程语言：Python
• 深度学习框架：PyTorch
• 目标检测库：mmdetection
• 图像处理库：kornia
• 关键技术：聚类算法、目标检测、图像分割

3. 安装和配置准备工作

在开始安装之前，请确保您的系统满足以下要求：

• Python版本：3.8或更高
• PyTorch版本：1.7.0或更高
• mmdetection版本：2.26.0（推荐），兼容版本为2.17.0到3.0.0之间
• kornia版本：0.6.9

安装步骤

1. 安装Python依赖 确保您的Python环境已经安装好，接下来需要安装项目的Python依赖。您可以使用pip进行安装：

   pip install -r requirements.txt
   

   如果requirements.txt文件不存在，则需要手动安装以下依赖：

   pip install torch torchvision
   pip install mmcv
   pip install mmdet
   pip install kornia
   

2. 准备数据集 下载并准备您所需的数据集。项目可以使用VisDrone数据集。您需要使用gen_crop.py脚本来生成训练数据集：

   python gen_crop.py
   

   对于验证数据集，您可以直接使用项目提供的annotations/VisDrone2019-DET_val_coco.json。

3. 训练模型

   • 单GPU训练：

     python train.py configs/yolc.py
     

   • 多GPU训练：

     ./dist_train.sh configs/yolc.py <your_gpu_num>
     

     替换<your_gpu_num>为实际使用的GPU数量。

4. 评估模型 使用以下命令来评估模型：

   python eval_yolc.py
   

   请确保设置saved_crop参数为0、1、2或3，以匹配LSM（k=saved_crop）。

按照以上步骤操作，您应该能够成功安装和配置YOLC项目。如果遇到任何问题，请参考项目的官方文档或向社区寻求帮助。