文件名称:ttach:使用PyTorch进行图像测试时间增强!
文件大小:15KB
文件格式:ZIP
更新时间:2024-06-12 21:58:44
computer-vision deep-learning pytorch classification segmentation
TTAch 使用PyTorch进行图像测试时间增强! 与“数据增强”对训练集所做的类似,“测试时间增强”的目的是对测试图像进行随机修改。 因此,我们不会对训练后的模型仅显示一次常规的“干净”图像,而是将其多次显示。 然后,我们将平均每个对应图像的预测,并将其作为我们的最终猜测[ ]。 Input | # input batch of images / / /|\ \ \ # apply augmentations (flips, rotation, scale, etc.) | | | | | | | # pass augmented batches through model | | | | | | | # reverse transfo
【文件预览】:
ttach-master
----setup.py(4KB)
----.gitignore(1KB)
----requirements.txt(0B)
----__init__.py(21B)
----MANIFEST.in(42B)
----LICENSE(1KB)
----Dockerfile.dev(163B)
----ttach()
--------transforms.py(8KB)
--------__init__.py(270B)
--------__version__.py(63B)
--------wrappers.py(6KB)
--------base.py(5KB)
--------functional.py(3KB)
--------aliases.py(809B)
----.github()
--------workflows()
----.deepsource.toml(84B)
----README.md(5KB)
----tests()
--------test_transforms.py(4KB)
--------test_base.py(1KB)