Skip to content

Arctitico/DDH

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

使用步骤

确保已安装conda:

conda --version

初始化环境并安装依赖:

conda env create -f environment.yml
conda activate DDH

将数据集放成以下结构:

mkdir -p datasets
datasets/
├── 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 val
python val.py --config config.yaml --checkpoint models/best.pt --split test

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages