文件名称:AugMix_TF2:用于Augmix纸的TF2.0端口
文件大小:12KB
文件格式:ZIP
更新时间:2024-05-13 20:00:24
python computer-vision deep-learning tensorflow classification
AugMix 是一种新的数据处理技术,可混合增强图像并执行增强图像的一致嵌入,从而提高了鲁棒性并改善了不确定性校准。 与其他增强技术相比,该技术获得了更好的结果。 它不仅提高了模型的准确性,而且还有助于提高模型的鲁棒性。 在PyTorch中。 这只是从PyTorch到Tensorflow 2.0的移植而已。 我使用ResNet20作为该模型的示例,但是您可以使用任何喜欢的模型。 要求 的Python> = 3.6.x numpy的> = 1.17 枕头> = 6.2 张量流== 2.0 用法: 我总是建议您使用anaconda来管理您的环境,但是您也可以使用virtualenv或直接安装软件包。 这是你的选择。 克隆仓库。 pip install requirements.txt python main.py --batch_size=128 --epochs=100 去做:
【文件预览】:
AugMix_TF2-master
----models()
--------__init__.py(0B)
--------resnet20.py(5KB)
----augmentation.py(3KB)
----main.py(4KB)
----utils.py(4KB)
----LICENSE(1KB)
----trainer.py(8KB)
----requirements.txt(62B)
----data_generator.py(3KB)
----config.py(122B)
----README.md(2KB)