确保已安装conda:
conda --version初始化环境并安装依赖:
conda env create -f environment.yml
conda activate DDH将数据集放成以下结构:
mkdir -p datasetsdatasets/
├── train/
│ ├── images/
│ └── labels/
├── val/
│ ├── images/
│ └── labels/
├── test/
│ ├── images/
│ └── labels/
遮盖敏感信息:
python mask_sensitive_info.py训练模型:
python train.py --config config.yaml在验证集和测试集上评估模型性能:
python val.py --config config.yaml --checkpoint models/best.pt --split valpython val.py --config config.yaml --checkpoint models/best.pt --split test