Kornia可微分計(jì)算機(jī)視覺庫

Kornia 是一個(gè)基于 PyTorch 的可微分計(jì)算機(jī)視覺庫。為了兼顧傳統(tǒng)視覺處理與深度學(xué)習(xí)的需求，Kornia 實(shí)現(xiàn)了：

1. 可微分的基礎(chǔ)計(jì)算機(jī)視覺算子。

2. 可微分的數(shù)據(jù)增廣（differentiable data augmentation）。

Kornia 基于 PyTorch，具備如下特性：

1. 可微分性。所有算子的梯度都可以通過 PyTorch 的 AutoGrad 計(jì)算，并使用 PyTorch 的優(yōu)化器（如：Adam）來優(yōu)化。

2. GPU/TPU 加速。除 CPU 外，Kornia 可以在 GPU 甚至 TPU 中進(jìn)行運(yùn)算。

3. 批數(shù)據(jù)處理。同時(shí)處理大量數(shù)據(jù)來提高運(yùn)行效率。