HOGGenerator¶
- class mmpretrain.models.selfsup.HOGGenerator(nbins=9, pool=8, gaussian_window=16)[源代码]¶
Generate HOG feature for images.
This module is used in MaskFeat to generate HOG feature. The code is modified from file slowfast/models/operators.py. Here is the link of HOG wikipedia.
- 参数:
- forward(x)[源代码]¶
Generate hog feature for each batch images.
- 参数:
x (torch.Tensor) – Input images of shape (N, 3, H, W).
- 返回:
Hog features.
- 返回类型: