#!/usr/bin/env python
|
# -*- coding: utf-8 -*-
|
# @Date : 2021-04-29 18:20:17
|
# @Author : Scheaven (snow_mail@foxmail.com)
|
# @Link : www.github.com
|
# @Version : $Id$
|
|
import torch
|
import sys
|
sys.path.append('.')
|
|
from data.data_utils import read_image
|
from predictor import ReID_Model
|
from config import get_cfg
|
from data.transforms.build import build_transforms
|
from engine.defaults import default_argument_parser, default_setup
|
import time
|
|
def setup(args):
|
"""
|
Create configs and perform basic setups.
|
"""
|
cfg = get_cfg()
|
cfg.merge_from_file(args.config_file)
|
cfg.merge_from_list(args.opts)
|
cfg.freeze()
|
default_setup(cfg, args)
|
return cfg
|
|
if __name__ == '__main__':
|
args = default_argument_parser().parse_args()
|
cfg = setup(args)
|
cfg.defrost()
|
cfg.MODEL.BACKBONE.PRETRAIN = False
|
|
model = ReID_Model(cfg)
|
|
|
model.torch2onnx()
|
