コード例 #1
0
ファイル: ucn_trainer.py プロジェクト: zeta1999/open-ucn 
    def __init__(
        self,
        config,
        data_loader,
        val_data_loader=None,
        test_data_loader=None,
    ):

        Trainer.__init__(self, config, data_loader, val_data_loader)
        self.best_val_metric = 'hit_ratio'
        self.train_max_iter = config.train_max_iter
        self.sift = cv2.xfeatures2d.SIFT_create()
コード例 #2
0
ファイル: tester.py プロジェクト: z80020100/OverlapPredator 
 def __init__(self, args):
     Trainer.__init__(self, args)