Exemplo n.º 1
0
def parse_args():
    parser = argparse.ArgumentParser(description="Train deepim network")
    # general
    parser.add_argument("--cfg",
                        help="experiment configure file name",
                        required=True,
                        type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    # training
    parser.add_argument(
        "--frequent",
        help="frequency of logging",
        default=config.default.frequent,
        type=int,
    )
    parser.add_argument("--gpus",
                        help="specify the gpu to be use",
                        required=True,
                        type=str)
    parser.add_argument("--temp",
                        help="turn on visualization",
                        action="store_true")
    parser.add_argument("--vis",
                        help="turn on visualization",
                        action="store_true")
    args = parser.parse_args()
    return args
Exemplo n.º 2
0
def parse_args():
    parser = argparse.ArgumentParser(description='Train deepim network')
    # general
    parser.add_argument('--cfg',
                        help='experiment configure file name',
                        required=True,
                        type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    # training
    parser.add_argument('--frequent',
                        help='frequency of logging',
                        default=config.default.frequent,
                        type=int)
    parser.add_argument('--gpus',
                        help='specify the gpu to be use',
                        required=True,
                        type=str)
    parser.add_argument('--temp',
                        help='turn on visualization',
                        action='store_true')
    parser.add_argument('--vis',
                        help='turn on visualization',
                        action='store_true')
    args = parser.parse_args()
    return args
def parse_args():
    """
    :return: parsed arguments and (updated) config.cfg object
    """
    parser = argparse.ArgumentParser(
        description='Train text recognition network')
    # general
    parser.add_argument(
        '--cfg',
        help='experiment configure file name',
        default='/home/citybuster/Projects/CRNN_STN_SEQ/experiments'
        '/synth90k/shadownet/normal_sts_shadownet_sgd_exponential-lr.yaml',
        type=str)

    args, rest = parser.parse_known_args()

    # update config
    update_config(args.cfg)
    # training
    parser.add_argument('--gpus', help='gpus', type=str)
    parser.add_argument('--workers',
                        help='num of dataloader workers',
                        type=int)

    args = parser.parse_args()

    return args
Exemplo n.º 4
0
def parse_args():
    parser = argparse.ArgumentParser(description='Train deeplab network')
    # general
    parser.add_argument('--cfg',
                        help='experiment configure file name',
                        required=True,
                        type=str)
    parser.add_argument('--backward-do-mirror',
                        dest='backward_do_mirror',
                        help='True means less gpu memory usage.',
                        default=False,
                        action='store_true')

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    if args.backward_do_mirror:
        os.environ['MXNET_BACKWARD_DO_MIRROR'] = '1'

    # training
    parser.add_argument('--frequent',
                        help='frequency of logging',
                        default=config.default.frequent,
                        type=int)
    args = parser.parse_args()
    return args
Exemplo n.º 5
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a Deeplab Network')
    # general
    parser.add_argument('--cfg',
                        help='experiment configure file name',
                        required=True,
                        type=str)
    parser.add_argument('--backward-do-mirror',
                        dest='backward_do_mirror',
                        help='True means less gpu memory usage.',
                        default=False,
                        action='store_true')
    args, rest = parser.parse_known_args()

    if args.backward_do_mirror:
        os.environ['MXNET_BACKWARD_DO_MIRROR'] = '1'
    update_config(args.cfg)

    # testing
    parser.add_argument('--vis',
                        help='turn on visualization',
                        action='store_true')
    parser.add_argument('--ignore_cache',
                        help='ignore cached results boxes',
                        action='store_true')
    parser.add_argument('--shuffle',
                        help='shuffle data on visualization',
                        action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 6
0
def parse_args():
    parser = argparse.ArgumentParser(description="Test a DeepIM Network")
    # general
    parser.add_argument(
        "--cfg", help="experiment configure file name", required=True, type=str
    )

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # testing
    parser.add_argument("--vis", help="turn on visualization", action="store_true")
    parser.add_argument(
        "--vis_video", help="turn on video visualization", action="store_true"
    )
    parser.add_argument(
        "--vis_video_zoom", help="turn on zoom video visualization", action="store_true"
    )
    parser.add_argument(
        "--ignore_cache",
        help="ignore cached pose prediction results",
        action="store_true",
    )
    parser.add_argument(
        "--gpus", help="specify the gpu to be use", required=True, type=str
    )
    parser.add_argument(
        "--skip_flow", help="whether skip flow during test", action="store_true"
    )
    args = parser.parse_args()
    return args
Exemplo n.º 7
0
def parse_args():
    parser = argparse.ArgumentParser(description='train detnet network')
    parser.add_argument('--cfg',help='configure file name',type = str, default = './cfgs/detnet.yaml')
    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    return args
def parse_args():
    # print('test----parse_args----in----')
    parser = argparse.ArgumentParser(description='Test a R-FCN network')
    # general
    # 通常测试一个网络都是使用cfg文件yaml
    parser.add_argument('--cfg',
                        help='experiment configure file name',
                        required=True,
                        type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # rcnn
    # rcnn相应参数,包括vis可视化,ignore_cache忽视缓存,thresh有效检测的阈值,shuffle可视化中随机数据
    parser.add_argument('--vis',
                        help='turn on visualization',
                        action='store_true')
    parser.add_argument('--ignore_cache',
                        help='ignore cached results boxes',
                        action='store_true')
    parser.add_argument('--thresh',
                        help='valid detection threshold',
                        default=1e-3,
                        type=float)
    parser.add_argument('--shuffle',
                        help='shuffle data on visualization',
                        action='store_true')
    args = parser.parse_args()
    # print('test----parse_args----out----')
    return args
Exemplo n.º 9
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a DeepIM Network')
    # general
    parser.add_argument(
        '--cfg',
        help='experiment configure file name',
        required=True,
        type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # testing
    parser.add_argument(
        '--vis', help='turn on visualization', action='store_true')
    parser.add_argument(
        '--vis_video', help='turn on video visualization', action='store_true')
    parser.add_argument(
        '--vis_video_zoom',
        help='turn on zoom video visualization',
        action='store_true')
    parser.add_argument(
        '--ignore_cache',
        help='ignore cached pose prediction results',
        action='store_true')
    parser.add_argument(
        '--gpus', help='specify the gpu to be use', required=True, type=str)
    parser.add_argument(
        '--skip_flow',
        help='whether skip flow during test',
        action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 10
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a Faster R-CNN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # rcnn
    parser.add_argument('--vis', help='turn on visualization', action='store_true')
    parser.add_argument('--ignore_cache', help='ignore cached results boxes', action='store_true')
    parser.add_argument('--thresh', help='valid detection threshold', default=1e-3, type=float)
    parser.add_argument('--shuffle', help='shuffle data on visualization', action='store_true')
    parser.add_argument('--test_epoch', help='the epoch model to be test', default=config.TEST.test_epoch, type=int)
    # nms
    parser.add_argument('--nms', help='params for nms or softnms', default=config.TEST.NMS, type=float)
    parser.add_argument('--softnms', help='whether to enable softnms', default=config.TEST.SOFTNMS, action='store_true')
    parser.add_argument('--naive_nms', help='whether to enable naive nms', default=False, action='store_true')
    parser.add_argument('--first_n', help='first_n for learn nms or nms', default=config.TEST.FIRST_N, type=int)
    parser.add_argument('--merge', help='merge method for learn nms', default=config.TEST.MERGE_METHOD, type=int)
    parser.add_argument('--debug', help='whether to enable debug mode', default=False, action='store_true')
    # dataset
    parser.add_argument('--test_set', help='which set to be tested', default=config.dataset.test_image_set, type=str)
    args, rest = parser.parse_known_args()
    # update config
    config.TEST.test_epoch = args.test_epoch
    config.TEST.NMS = args.nms
    config.TEST.SOFTNMS = args.softnms and (not args.naive_nms)
    config.TEST.FIRST_N = args.first_n
    config.TEST.MERGE_METHOD = args.merge
    config.dataset.test_image_set = args.test_set
    return args
def parse_args():
    """
    :return: parsed arguments and (updated) config.cfg object
    """
    parser = argparse.ArgumentParser(
        description='Train text recognition network')
    # general
    parser.add_argument(
        '--cfg',
        help='experiment configure file name',
        default='/home/citybuster/Projects/CRNN_STN_SEQ/experiments/'
        'synth90k/shadownet/stn_ctc_shadownet_sgd_exponential-lr.yaml',
        type=str)

    args, rest = parser.parse_known_args()

    # update config
    update_config(args.cfg)
    parser.add_argument(
        '--data',
        help='The raw dataset dir',
        default='/home/citybuster/Data/mnt/ramdisk/max/90kDICT32px',
        type=str)

    args = parser.parse_args()
    return args
Exemplo n.º 12
0
Arquivo: test.py Projeto: lyj96/STSN
def parse_args():
    parser = argparse.ArgumentParser(description='Test a R-FCN network')
    # general
    parser.add_argument('--cfg',
                        help='experiment configure file name',
                        required=True,
                        type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # rcnn
    parser.add_argument('--vis',
                        help='turn on visualization',
                        action='store_true')
    parser.add_argument('--ignore_cache',
                        help='ignore cached results boxes',
                        action='store_true')
    parser.add_argument('--thresh',
                        help='valid detection threshold',
                        default=1e-3,
                        type=float)
    parser.add_argument('--shuffle',
                        help='shuffle data on visualization',
                        action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 13
0
def parse_args():
    parser = argparse.ArgumentParser(description='Deploy a Faster R-CNN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    args = parser.parse_args()
    return args
Exemplo n.º 14
0
def parse_args():
    parser = argparse.ArgumentParser(description='Train PIX2PIX')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    args = parser.parse_args()
    return args
Exemplo n.º 15
0
def parse_args():
    parser = argparse.ArgumentParser(description='demo of detnet network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', default='./cfgs/detnet.yaml', type=str)

    args, rest = parser.parse_known_args()

    parser.add_argument('--thresh', help='valid detection threshold', default=0.5, type=float)
    parser.add_argument('--image', help='image path', default='./images', type=str)
    args = parser.parse_args()
    update_config(args.cfg)
    return args
def main():
    from utils.debug import set_working_dir
    # set the working directory as appropriate
    set_working_dir()

    args = parse_args()

    print('Called with argument:', args)
    # update config
    update_config(args.cfg)
    check_config(config)
    train()
Exemplo n.º 17
0
def parse_args():
    parser = argparse.ArgumentParser(description='Train FCIS Network')
    # general
    # configuration file is required
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(cur_path + '/../experiments/fcis/cfgs/' +args.cfg)

    args = parser.parse_args()
    return args
Exemplo n.º 18
0
def init_detect_model():
    # get symbol
    pprint.pprint(config)
    update_config(RFCN_DCN_CONFIG["config_yaml_file"])
    config.symbol = 'resnet_v1_101_rfcn_dcn'
    sym_instance = eval(config.symbol + '.' + config.symbol)()
    sym = sym_instance.get_symbol(config, is_train=False)
    arg_params, aux_params = load_param(os.path.join(
        RFCN_DCN_CONFIG['modelParam']['modelBasePath'], 'rfcn_voc'),
                                        RFCN_DCN_CONFIG['modelParam']['epoch'],
                                        process=True)
    return [sym, arg_params, aux_params]
Exemplo n.º 19
0
def parse_args():
    parser = argparse.ArgumentParser(description='Show Flow-Guided Feature Aggregation demo')

    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    args = parser.parse_args()
    return args
Exemplo n.º 20
0
def parse_args():
    parser = argparse.ArgumentParser(description='Train Faster-RCNN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    # training
    parser.add_argument('--frequent', help='frequency of logging', default=config.default.frequent, type=int)
    args = parser.parse_args()
    return args
Exemplo n.º 21
0
def parse_args():
    parser = argparse.ArgumentParser(description='Train R-FCN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', default="../experiments/fgfa_rfcn/cfgs/resnet_v1_101_flownet_imagenet_vid_rfcn_end2end_ohem.yaml",type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    # training
    parser.add_argument('--frequent', help='frequency of logging', default=config.default.frequent, type=int)
    args = parser.parse_args()
    return args
def parse_args():
    parser = argparse.ArgumentParser(description='Train Faster-RCNN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)

    # training
    parser.add_argument('--frequent', help='frequency of logging', default=config.default.frequent, type=int)
    args, rest = parser.parse_known_args()
    return args
Exemplo n.º 23
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a Faster R-CNN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # rcnn
    parser.add_argument('--vis', help='turn on visualization', action='store_true')
    parser.add_argument('--thresh', help='valid detection threshold', default=1e-3, type=float)
    parser.add_argument('--use_box_voting', help='use box voting in test', action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 24
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a Deeplab Network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # testing
    parser.add_argument('--vis', help='turn on visualization', action='store_true')
    parser.add_argument('--ignore_cache', help='ignore cached results boxes', action='store_true')
    parser.add_argument('--shuffle', help='shuffle data on visualization', action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 25
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a Deeplab Network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # testing
    parser.add_argument('--vis', help='turn on visualization', action='store_true')
    parser.add_argument('--ignore_cache', help='ignore cached results boxes', action='store_true')
    parser.add_argument('--shuffle', help='shuffle data on visualization', action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 26
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a R-FCN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # rcnn
    parser.add_argument('--vis', help='turn on visualization', action='store_true')
    parser.add_argument('--ignore_cache', help='ignore cached results boxes', action='store_true')
    parser.add_argument('--thresh', help='valid detection threshold', default=1e-4, type=float)
    parser.add_argument('--shuffle', help='shuffle data on visualization', action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 27
0
def parse_args():
    parser = argparse.ArgumentParser(description='Test a R-FCN network')
    # general
    parser.add_argument('--cfg',
                        help='experiment configure file name',
                        required=True,
                        type=str)
    parser.add_argument('--usePhilly',
                        help='experiment is on the philly',
                        action='store_true')

    args, rest = parser.parse_known_args()
    update_config(args.cfg)
    config.USE_PHILLY = args.usePhilly

    # modfication for philly
    if config.USE_PHILLY:
        parser.add_argument('--dataDir',
                            help='input directory for Philly jobs',
                            required=True,
                            type=str)
        parser.add_argument('--modelDir',
                            help='output directory for Philly jobs',
                            required=True,
                            type=str)
        args, rest = parser.parse_known_args()
        update_philly_config(args.modelDir, args.dataDir)

    # rcnn
    parser.add_argument('--vis',
                        help='turn on visualization',
                        action='store_true')
    parser.add_argument('--show_gt',
                        help='turn on visualization on gt',
                        action='store_true')
    parser.add_argument('--ignore_cache',
                        help='ignore cached results boxes',
                        action='store_true')
    parser.add_argument('--thresh',
                        help='valid detection threshold',
                        default=1e-4,
                        type=float)
    parser.add_argument('--shuffle',
                        help='shuffle data on visualization',
                        action='store_true')
    args = parser.parse_args()
    return args
Exemplo n.º 28
0
def main():
    args = parse_args()
    print 'Called with argument:', args
    cfg_path = args.cfg
    update_config(cfg_path)

    # create logger
    logger, output_path = create_logger(config.output_path, cfg_path,
                                        config.dataset.image_set)

    # print config
    pprint.pprint(config)
    logger.info('training config: {}\n'.format(pprint.pformat(config)))

    # train_net(cfg_path, ctx, config.network.pretrained, config.network.pretrained_epoch,
    #           config.TRAIN.model_prefix, config.TRAIN.begin_epoch, config.TRAIN.end_epoch,
    #           config.TRAIN.lr, config.TRAIN.lr_step)
    train_net(config, output_path, logger)
Exemplo n.º 29
0
def main():
    update_config(os.path.join(cur_path, 'demo/models', args.configYamlFile))
    if args.visualize:
        if not args.visualizeOutPutPath:
            args.visualizeOutPutPath = os.path.join(cur_path,
                                                    'visualizeOutPutPath')
        if not os.path.exists(args.visualizeOutPutPath):
            os.makedirs(args.visualizeOutPutPath)
    if not os.path.exists(os.path.split(args.outputFilePath)[0]):
        os.makedirs(os.path.split(args.outputFilePath)[0])
    fileOp = open(args.outputFilePath + "-" + str(args.epoch) + "-threshold-" +
                  str(args.threshold), 'a+')  # 追加的方式,如果不存在就创建
    need_process_images_path_list = []
    if args.localImageListFile:
        basePath = ''
        if args.localImageBasePath:
            basePath = args.localImageBasePath
        with open(args.localImageListFile, 'r') as f:
            for line in f.readlines():
                line = line.strip()
                if len(line) == 0:
                    continue
                if line.endswith('.jpg'):
                    pass
                else:
                    line = line + '.jpg'
                image_path = os.path.join(basePath, line)
                need_process_images_path_list.append(image_path)
        process_image_fun(isUrlFlag=False,
                          imagesPath=need_process_images_path_list,
                          fileOp=fileOp,
                          vis=args.visualize)
    elif args.urlImageListFile:
        with open(args.urlImageListFile, 'r') as f:
            for line in f.readlines():
                line = line.strip()
                if len(line) == 0:
                    continue
                need_process_images_path_list.append(line)
        process_image_fun(isUrlFlag=True,
                          imagesPath=need_process_images_path_list,
                          fileOp=fileOp,
                          vis=args.visualize)
    pass
def parse_args():
    parser = argparse.ArgumentParser(description='Test a Faster R-CNN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)

    args, rest = parser.parse_known_args()
    update_config(args.cfg)

    # rcnn
    parser.add_argument('--vis', help='turn on visualization', action='store_true')
    parser.add_argument('--ignore_cache', help='ignore cached results boxes', action='store_true')
    parser.add_argument('--thresh', help='valid detection threshold', default=1e-3, type=float)
    parser.add_argument('--shuffle', help='shuffle data on visualization', action='store_true')
    parser.add_argument('--sample-stride', help='sample stride', default=-1, type=int)
    parser.add_argument('--key-frame-interval', help='key frame interval', default=-1, type=int)
    parser.add_argument('--video-shuffle', help='video shuffle', action='store_true')
    parser.add_argument('--test-pretrained', help='test pretrained model', type=str)
    args = parser.parse_args()
    return args
 def parse_args():
     parser = argparse.ArgumentParser(description='Test a R-FCN network')
     # general
 #    parser.add_argument('--cfg', help='experiment configure file name', required=False, type=str)
     parser.add_argument('--cfg', help='experiment configure file name', 
                 default='../experiments/fgfa_rfcn/cfgs/resnet_v1_101_flownet_imagenet_vid_rfcn_end2end_ohem.yaml', required=False, type=str)
 
     args, rest = parser.parse_known_args()
     update_config(args.cfg)
 
     # rcnn
     parser.add_argument('--orig_pred', help='True to get and evaluate original predictions. False to get predictions suited for REPP', 
                         action='store_true')
     parser.add_argument('--vis', help='turn on visualization', action='store_true')
     parser.add_argument('--ignore_cache', help='ignore cached results boxes', action='store_true')
     parser.add_argument('--thresh', help='valid detection threshold', default=1e-3, type=float)
     parser.add_argument('--shuffle', help='shuffle data on visualization', action='store_true')
     parser.add_argument('--dataset_path', help='path of the Imagenet VID dataset', type=str)
     args = parser.parse_args()
     return args
Exemplo n.º 32
0
def parse_args():
    parser = argparse.ArgumentParser(description='Train R-FCN network')
    # general
    parser.add_argument('--cfg', help='experiment configure file name', required=True, type=str)
    parser.add_argument('--usePhilly', help='experiment is on the philly', action='store_true')

    args, rest = parser.parse_known_args()
    # update config
    update_config(args.cfg)
    config.USE_PHILLY = args.usePhilly

    # modfication for philly
    if config.USE_PHILLY:
        parser.add_argument('--dataDir', help='input directory for Philly jobs', required=True, type=str)
        parser.add_argument('--modelDir', help='output directory for Philly jobs', required=True, type=str)
        args, rest = parser.parse_known_args()
        update_philly_config(args.modelDir, args.dataDir)

    # training
    parser.add_argument('--frequent', help='frequency of logging', default=config.default.frequent, type=int)
    args = parser.parse_args()
    return args
def main():
    update_config(RFCN_DCN_CONFIG["config_yaml_file"])
    time_str = time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime())
    visualizeOutPutPath = None
    if int(args.visualizeFlag) == 1:
        visualizeOutPutPath = os.path.join(os.path.dirname(args.imageListFile),
                                           "visDraw-" + time_str)
        if not os.path.exists(visualizeOutPutPath):
            os.makedirs(visualizeOutPutPath)
    outputFilePath = args.imageListFile + "-result-" + str(
        args.outputFileFlag) + '-' + time_str
    fileOp = open(outputFilePath, 'a+')  # 追加的方式,如果不存在就创建
    need_process_images_path_list = []
    with open(args.imageListFile, 'r') as f:
        for line in f.readlines():
            line = line.strip()
            if len(line) == 0:
                continue
            need_process_images_path_list.append(line)
    process_image_fun(urlFlag=args.urlFlag,
                      imagesPath=need_process_images_path_list,
                      fileOp=fileOp,
                      vis=visualizeOutPutPath)
Exemplo n.º 34
0
import os
import sys
import logging
import pprint
import cv2
from config.config import config, update_config
from utils.image import resize, transform
from PIL import Image
import numpy as np

# get config
os.environ['PYTHONUNBUFFERED'] = '1'
os.environ['MXNET_CUDNN_AUTOTUNE_DEFAULT'] = '0'
os.environ['MXNET_ENABLE_GPU_P2P'] = '0'
cur_path = os.path.abspath(os.path.dirname(__file__))
update_config(cur_path + '/../experiments/deeplab/cfgs/deeplab_cityscapes_demo.yaml')

sys.path.insert(0, os.path.join(cur_path, '../external/mxnet', config.MXNET_VERSION))
import mxnet as mx
from core.tester import pred_eval, Predictor
from symbols import *
from utils.load_model import load_param
from utils.tictoc import tic, toc

def parse_args():
    parser = argparse.ArgumentParser(description='Show Deformable ConvNets demo')
    # general
    parser.add_argument('--deeplab_only', help='whether use Deeplab only (w/o Deformable ConvNets)', default=False, action='store_true')

    args = parser.parse_args()
    return args
Exemplo n.º 35
0
import argparse
import os
import sys
import logging
import pprint
import cv2
from config.config import config, update_config
from utils.image import resize, transform
import numpy as np
# get config
os.environ['PYTHONUNBUFFERED'] = '1'
os.environ['MXNET_CUDNN_AUTOTUNE_DEFAULT'] = '0'
os.environ['MXNET_ENABLE_GPU_P2P'] = '0'
cur_path = os.path.abspath(os.path.dirname(__file__))
update_config(cur_path + '/../experiments/rfcn/cfgs/rfcn_coco_demo.yaml')

sys.path.insert(0, os.path.join(cur_path, '../external/mxnet', config.MXNET_VERSION))
import mxnet as mx
from core.tester import im_detect, Predictor
from symbols import *
from utils.load_model import load_param
from utils.show_boxes import show_boxes
from utils.tictoc import tic, toc
from nms.nms import py_nms_wrapper, cpu_nms_wrapper, gpu_nms_wrapper

def parse_args():
    parser = argparse.ArgumentParser(description='Show Deformable ConvNets demo')
    # general
    parser.add_argument('--rfcn_only', help='whether use R-FCN only (w/o Deformable ConvNets)', default=False, action='store_true')
Exemplo n.º 36
0
import argparse
import os
import glob
import sys
import logging
import pprint
import cv2
from config.config import config, update_config
from utils.image import resize, transform
import numpy as np
# get config
os.environ['PYTHONUNBUFFERED'] = '1'
os.environ['MXNET_CUDNN_AUTOTUNE_DEFAULT'] = '0'
os.environ['MXNET_ENABLE_GPU_P2P'] = '0'
cur_path = os.path.abspath(os.path.dirname(__file__))
update_config(cur_path + '/../experiments/dff_rfcn/cfgs/dff_rfcn_vid_demo.yaml')

sys.path.insert(0, os.path.join(cur_path, '../external/mxnet', config.MXNET_VERSION))
import mxnet as mx
from core.tester import im_detect, Predictor
from symbols import *
from utils.load_model import load_param
from utils.show_boxes import show_boxes, draw_boxes
from utils.tictoc import tic, toc
from nms.nms import py_nms_wrapper, cpu_nms_wrapper, gpu_nms_wrapper

def parse_args():
    parser = argparse.ArgumentParser(description='Show Deep Feature Flow demo')
    args = parser.parse_args()
    return args
import _init_paths

import os
import sys
import pprint
import cv2
from config.config import config, update_config
from utils.image import resize, transform
import numpy as np
# get config
os.environ['PYTHONUNBUFFERED'] = '1'
os.environ['MXNET_CUDNN_AUTOTUNE_DEFAULT'] = '0'
os.environ['MXNET_ENABLE_GPU_P2P'] = '0'
cur_path = os.path.abspath(os.path.dirname(__file__))
update_config(cur_path + '/../experiments/rfcn/cfgs/deform_psroi_demo.yaml')

sys.path.insert(0, os.path.join(cur_path, '../external/mxnet', config.MXNET_VERSION))
import mxnet as mx
from core.tester import Predictor
from symbols import *
from utils.load_model import load_param
from utils.show_offset import show_dpsroi_offset

def main():
    # get symbol
    pprint.pprint(config)
    sym_instance = eval(config.symbol + '.' + config.symbol)()
    sym = sym_instance.get_symbol_rfcn(config, is_train=False)

    # load demo data
    image_names = ['000057.jpg', '000149.jpg', '000351.jpg', '002535.jpg']