|
1 | 1 | from .backbones import (ResNet, ResNet2Plus1d, ResNet3d, ResNet3dCSN,
|
2 | 2 | ResNet3dSlowFast, ResNet3dSlowOnly, ResNetTIN,
|
3 | 3 | ResNetTSM)
|
4 |
| -from .builder import (build_backbone, build_head, build_localizer, build_model, |
5 |
| - build_recognizer) |
| 4 | +from .builder import (build_backbone, build_head, build_localizer, build_loss, |
| 5 | + build_model, build_neck, build_recognizer) |
6 | 6 | from .common import Conv2plus1d
|
7 | 7 | from .heads import BaseHead, I3DHead, SlowFastHead, TPNHead, TSMHead, TSNHead
|
8 | 8 | from .localizers import BMN, PEM, TEM
|
|
20 | 20 | 'ResNet3dSlowFast', 'SlowFastHead', 'Conv2plus1d', 'ResNet3dSlowOnly',
|
21 | 21 | 'BCELossWithLogits', 'LOCALIZERS', 'build_localizer', 'PEM', 'TEM',
|
22 | 22 | 'BinaryLogisticRegressionLoss', 'BMN', 'BMNLoss', 'build_model',
|
23 |
| - 'OHEMHingeLoss', 'SSNLoss', 'ResNet3dCSN', 'ResNetTIN', 'TPN', 'TPNHead' |
| 23 | + 'OHEMHingeLoss', 'SSNLoss', 'ResNet3dCSN', 'ResNetTIN', 'TPN', 'TPNHead', |
| 24 | + 'build_loss', 'build_neck' |
24 | 25 | ]
|
0 commit comments