Python DistributedWrapper Examples

Example #1

import os
from operators.distributed_wrapper import DistributedWrapper

from utils.metrics.metrics import evaluate_results
# ==== import your configuration here ===
from configs.rrnet_config import Config
# ==== import your model operator here ===
from operators.rrnet_operator import RRNetOperator

if __name__ == "__main__":
    print("Start generating Txt file ...")
    dis_operator = DistributedWrapper(Config, RRNetOperator)
    dis_operator.eval()

    print('Start Evaluating ...')
    result_dir = Config.Val.result_dir
    gt_dir = os.path.join(Config.data_root, 'val', 'annotations')
    evaluate_results(result_dir, gt_dir)

Example #2

File: train.py Project: Cynicsss/udwt-Robot-Picking-C-fast

from configs.rrnet_config import Config
from operators.distributed_wrapper import DistributedWrapper
from operators.train.rrnet_operator import RRNetTrainOperator
from operators.train.ctnet_operator import CTNetTrainOperator


if __name__ == '__main__':
    op = RRNetTrainOperator if Config.Model.use_rr else CTNetTrainOperator
    dis_operator = DistributedWrapper(Config, op)
    dis_operator.train()
    print('Training is Done!')

Example #3

File: train.py Project: Cynicsss/udwt-Robot-Picking-C

from configs.rrnet_config import Config
from operators.distributed_wrapper import DistributedWrapper
from operators.rrnet_operator import RRNetOperator

if __name__ == '__main__':
    dis_operator = DistributedWrapper(Config, RRNetOperator)
    dis_operator.train()
    print('Training is Done!')

Example #4

File: deep_sort.py Project: Cynicsss/udwt-Robot-Picking-C

import torch
import time
import cv2
import os
from operators.distributed_wrapper import DistributedWrapper
from configs.rrnet_config import Config
import numpy as np
from operators.rrnet_operator import RRNetOperator
from deep_sort import DeepSort
from utils.functional import draw_bboxes

if __name__ == '__main__':

    print('***** generating bounding box!! *****')
    dis_operator = DistributedWrapper(Config, RRNetOperator)
    dis_operator.deep_sort()

    # deepsort_echinus = DeepSort('./original_ckpt.t7')
    # deepsort_holothurian = DeepSort('./original_ckpt.t7')
    # deepsort_scallop = DeepSort('./original_ckpt.t7')

    num = 0

    print('***** start deep_sort!! *****')
    # for txt in os.listdir('./results/'):
    '''
    for i in range(3700 ,4499 + 1):
        start = time.time()
        bbox_xywh_echinus = []
        bbox_xywh_holothurian = []
        bbox_xywh_scallop = []