Commit 267b2e4a authored by sunxx1's avatar sunxx1
Browse files

Merge branch 'zhanggzh-main-patch-94239' into 'main'

Update Keras/keras-cv/examples/training/classification/imagenet/training_history.json, README.md

See merge request dcutoolkit/deeplearing/dlexamples_new!50
parents ac26d1fb 3262adde
1 简介
该测试用例为keras版本分类网络测试,使用TFrecord格式的ImageNet数据集
2 环境部署要求
DTK22.04.2
TensorFlow-2.9.0
3 安装
执行 python setup.py build
python setup.py build
测试 进入python环境 执行import tensorflow as tf
import keras_cv
4 训练执行 train_keras_cv.sh脚本或将脚本中命令单独取出执行
\ No newline at end of file
{ {
"densenet121": { "densenet121": {
"v0": { "v0": {
"accelerators": 2, "accelerators": 4,
"args": { "args": {
"batch_size": "64" "batch_size": "64"
}, },
"contributor": "ianstenbit", "lr": "0.05",
"epochs_trained": 84, "epochs_trained": 90,
"script": { "script": {
"name": "basic_training.py", "name": "basic_training.py"
"version": "90d4c3548a2e989fe52d6cf7ae7439af794f0ae6"
}, },
"tensorboard_logs": "https://tensorboard.dev/experiment/K5Q0gAk0RayXwP0WsLPpMA/",
"validation_accuracy": "0.6771" "validation_accuracy": "0.7019"
} }
}, },
"densenet169": { "efficientnetv2s": {
"v0": { "v0": {
"accelerators": 2, "accelerators": 4,
"args": { "args": {
"batch_size": "64" "batch_size": "64",
}, "initial_learning_rate": "0.05"
"contributor": "ianstenbit",
"epochs_trained": 50,
"script": {
"name": "basic_training.py",
"version": "90d4c3548a2e989fe52d6cf7ae7439af794f0ae6"
},
"tensorboard_logs": "https://tensorboard.dev/experiment/aQIvxQEgTqajldKxp688Nw/",
"validation_accuracy": "0.6613"
}
},
"densenet201": {
"v0": {
"accelerators": 8,
"args": {
"batch_size": "512"
}, },
"contributor": "ianstenbit", "epochs_trained": 43,
"epochs_trained": 166,
"script": { "script": {
"name": "basic_training.py", "name": "basic_training.py"
"version": "b0b349612e00ab34c25af5467ddd3b48d6fbf7a3"
}, },
"tensorboard_logs": "https://tensorboard.dev/experiment/6iLPGz5RSEiyPymgzJbKIQ/", "validation_accuracy": "0.7405"
"validation_accuracy": "0.7469"
} }
}, },
"efficientnetv2b0": { "ResNet101": {
"v0": { "v0": {
"accelerators": 8, "accelerators": 4,
"args": { "args": {
"batch_size": "64", "batch_size": "64",
"initial_learning_rate": ".0125" "initial_learning_rate": "0.05"
}, },
"contributor": "ianstenbit", "epochs_trained": 81,
"epochs_trained": 320,
"script": { "script": {
"name": "basic_training.py", "name": "basic_training.py"
"version": "e349ca5563b05548996f438fa03b2f34a8231ca3"
}, },
"tensorboard_logs": "https://tensorboard.dev/experiment/kBs9YZkwQAeVNfv8JPKCLw/", "validation_accuracy": "0.7381"
"validation_accuracy": "0.7527"
} }
}, },
"efficientnetv2b1": { "DarkNet53": {
"v0": { "v0": {
"accelerators": 8, "accelerators": 4,
"args": { "args": {
"batch_size": "64", "batch_size": "64",
"initial_learning_rate": ".0125" "initial_learning_rate": "0.0125"
}, },
"contributor": "ianstenbit", "epochs_trained": 120,
"epochs_trained": 288,
"script": { "script": {
"name": "basic_training.py", "name": "basic_training.py"
"version": "e349ca5563b05548996f438fa03b2f34a8231ca3"
}, },
"tensorboard_logs": "https://tensorboard.dev/experiment/jQAQBh6LQUep18CDayP8ww/", "validation_accuracy": "0.7433"
"validation_accuracy": "0.7560"
} }
}, },
"efficientnetv2b2": { "RegNetX064": {
"v0": { "v0": {
"accelerators": 8, "accelerators": 4,
"args": { "args": {
"batch_size": "64", "batch_size": "64",
"initial_learning_rate": ".0125" "initial_learning_rate": "0.01"
}, },
"contributor": "ianstenbit", "epochs_trained": 107,
"epochs_trained": 313,
"script": { "script": {
"name": "basic_training.py", "name": "basic_training.py"
"version": "02b41ea91b972cdd29c27dbc4d79e6a0b4e90de2"
}, },
"tensorboard_logs": "https://tensorboard.dev/experiment/iyhN2qvIRrqj6C0Q328drg/", "validation_accuracy": "0.7506"
"validation_accuracy": "0.7699"
} }
}, },
"resnet50v2": { "resnet50v2": {
"v0": {
"accelerators": 2,
"args": {
"batch_size": "64",
"initial_learning_rate": "0.005"
},
"contributor": "ianstenbit",
"epochs_trained": 132,
"script": {
"name": "basic_training.py",
"version": "3288c3ab31ce1c35fe7505e245fdfa9c593af78e"
},
"tensorboard_logs": "https://tensorboard.dev/experiment/QlkKjMkqQxm3jbzOlzBvWA/",
"validation_accuracy": "0.6337"
},
"v1": {
"accelerators": 2,
"args": {
"batch_size": "128"
},
"contributor": "ianstenbit",
"epochs_trained": 168,
"script": {
"name": "basic_training.py",
"version": "8fcffd9ee81ca9892f73d8ec3ac0ba475d2f1426"
},
"tensorboard_logs": "https://tensorboard.dev/experiment/TQ5r1EhXS4SDDagBD84rgA/",
"validation_accuracy": "0.7550"
},
"v2": { "v2": {
"accelerators": 8, "accelerators": 4,
"args": { "args": {
"batch_size": "64", "batch_size": "64",
"initial_learning_rate": ".0125" "initial_learning_rate": ".0125"
}, },
"contributor": "ianstenbit", "epochs_trained": 120,
"epochs_trained": 150,
"script": { "script": {
"name": "basic_training.py", "name": "basic_training.py"
"version": "02b41ea91b972cdd29c27dbc4d79e6a0b4e90de2"
}, },
"tensorboard_logs": "https://tensorboard.dev/experiment/ReyWQHwETwah0nqlXl8BOA/", "validation_accuracy": "0.7387"
"validation_accuracy": "0.7612"
} }
},
"script_authors": {
"basic_training.py": [
"ianstenbit",
"DavidLandup0"
]
} }
} }
...@@ -15,6 +15,7 @@ DCU深度学习样例 ...@@ -15,6 +15,7 @@ DCU深度学习样例
| TensorFlow | v1.15 | Yes | FP32/FP16 | Yes | ResNet50/VGG16/Inception-V3/MobileNet-V2 | [classfication](http://10.0.50.24/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow/ComputeVision/Classification) | | TensorFlow | v1.15 | Yes | FP32/FP16 | Yes | ResNet50/VGG16/Inception-V3/MobileNet-V2 | [classfication](http://10.0.50.24/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow/ComputeVision/Classification) |
| TensorFlow | v2.7 | Yes | FP32/FP16 | Yes | ResNet50/VGG16/Inception-V3 | [classfication](http://10.0.50.24/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/Classification) | | TensorFlow | v2.7 | Yes | FP32/FP16 | Yes | ResNet50/VGG16/Inception-V3 | [classfication](http://10.0.50.24/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/TensorFlow2x/Classification) |
| PyTorch | v1.10 | Yes | FP32/FP16 | Yes | ResNet50/Vgg16/Inception-V3/MobileNet-V2 | [classfication](http://10.0.50.24/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Accuracy_Validation/ResNet50) | | PyTorch | v1.10 | Yes | FP32/FP16 | Yes | ResNet50/Vgg16/Inception-V3/MobileNet-V2 | [classfication](http://10.0.50.24/dcutoolkit/deeplearing/dlexamples_new/-/tree/main/PyTorch/Compute-Vision/Accuracy_Validation/ResNet50) |
| keras | v2.9 | Yes | FP32 | Yes | Densenet121/Efficientnetv2s/ResNet101/DarkNet53/RegNetX064/resnet50v2 | [classfication](http://10.0.50.24/dcutoolkit/deeplearing/dlexamples_new/-/blob/main/Keras/keras-cv/examples/training/classification/imagenet) |
<br> <br>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment