(config)
| 2 | |
| 3 | |
| 4 | def make_models(config): |
| 5 | if config.MODEL.NAME == "SphereVLAD": |
| 6 | if config.MODEL.TYPE == "LiSPH": |
| 7 | from .lidar.spherevlad import SphereVLAD |
| 8 | elif config.MODEL.TYPE == "Image": |
| 9 | from .visual.spherevlad import SphereVLAD |
| 10 | model = SphereVLAD(config) |
| 11 | elif config.MODEL.NAME == "SphereVLAD2": |
| 12 | from .lidar.spherevlad2 import SphereVLAD2 |
| 13 | model = SphereVLAD2(config) |
| 14 | else: |
| 15 | raise ValueError("Wrong Model Name!") |
| 16 | |
| 17 | if config.TRAINING.IS_TRAIN and config.WEIGHT.FREEZE_WEIGHT: |
| 18 | pass |
| 19 | return model |
no test coverage detected