<filename>src/lib/trains/train_factory.py<gh_stars>0
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from .ctdet import CtdetTrainer
from .ddd import DddTrainer
from .exdet import ExdetTrainer
from .multi_pose import MultiPoseTrainer
from .ctdet_iter import CtdetTrainerIter
from .ctdet_tracking import CtdetTracking

train_factory = {
  'exdet': ExdetTrainer, 
  'ddd': DddTrainer,
  'ctdet': CtdetTrainer,
  'multi_pose': MultiPoseTrainer,
  'ctdet_stream': CtdetTrainerIter,
  'ctdet_semseg': CtdetTrainerIter,
  'ctdet_tracking': CtdetTracking
}
<filename>src/lib/trains/train_factory.py<gh_stars>0
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from .ctdet import CtdetTrainer
from .ddd import DddTrainer
from .exdet import ExdetTrainer
from .multi_pose import MultiPoseTrainer
from .ctdet_iter import CtdetTrainerIter
from .ctdet_tracking import CtdetTracking

train_factory = {
  'exdet': ExdetTrainer, 
  'ddd': DddTrainer,
  'ctdet': CtdetTrainer,
  'multi_pose': MultiPoseTrainer,
  'ctdet_stream': CtdetTrainerIter,
  'ctdet_semseg': CtdetTrainerIter,
  'ctdet_tracking': CtdetTracking
}