Skip to content

XavierYorke/MyAttaNet

Repository files navigation

AttaNet模型的PyTorch版本复现

配置文件

config.yaml: 参数设置

数据集的准备

文件夹格式

├─data_dir
│  ├─images
│  │  └─train
│  └─labels
│      └─train

训练模型

python main.py

评价系数

python evaluate.py

导出模型为ONNX

python export.py

推理

python inference.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages