Python-图像分类目标检测姿态估计分割的Pytorch实现

Repo for most of CV problems, such as image classification, object detection, pose estimation, segmentation, and so on.

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author: Donny You（youansheng@gmail.com）
# Main scripts for computer vision.


import os
import json
import time
import random
import argparse
import torch
import torch.backends.cudnn as cudnn

from runner.runner_selector import RunnerSelector
from runner.tools.controller import Controller
from tools.util.configer import Configer
from tools.util.logger import Logger as Log


def str2bool（v）:
    “““ Usage:
    parser.add_argument（‘--pretrained‘ type=str2bool nargs=‘?‘ const=True
                        dest=‘pretrained‘ help=‘Whether to use pretrained models.‘）
    “““
    if v.lower（） in （‘yes‘ ‘true‘ ‘t‘ ‘y‘ ‘1‘）:
        return True
    elif v.lower（） in （‘no‘ ‘false‘ ‘f‘ ‘n‘ ‘0‘）:
        return False
    else:
        raise argparse.ArgumentTypeError（‘Unsupported value encountered.‘）


if __name__ == “__main__“:
    parser = argparse.ArgumentParser（）
    parser.add_argument（‘--config_file‘ default=None type=str
                        dest=‘config_file‘ help=‘The file of the hyper parameters.‘）
    parser.add_argument（‘--phase‘ default=‘train‘ type=str
                        dest=‘phase‘ help=‘The phase of module.‘）
    parser.add_argument（‘--gpu‘ default=[0 1 2 3] nargs=‘+‘ type=int
                        dest=‘gpu‘ help=‘The gpu list used.‘）

    # ***********  Params for data.  **********
    parser.add_argument（‘--data_dir‘ default=None type=str
                        dest=‘data.data_dir‘ help=‘The Directory of the data.‘）
    parser.add_argument（‘--tag‘ default=None type=str
                        dest=‘data.tag‘ help=‘The Tag of the data.‘）
    parser.add_argument（‘--include_val‘ type=str2bool nargs=‘?‘ default=False
                        dest=‘data.include_val‘ help=‘Include validation set for final training.‘）
    parser.add_argument（‘--drop_last‘ type=str2bool nargs=‘?‘ default=False
                        dest=‘data.drop_last‘ help=‘Fix bug for syncbn.‘）
    parser.add_argument（‘--workers‘ default=None type=int
                        dest=‘data.workers‘ help=‘The number of workers to load data.‘）
    parser.add_argument（‘--train_batch_size‘ default=None type=int
                        dest=‘train.batch_size‘ help=‘The batch size of training.‘）
    parser.add_argument（‘--val_batch_size‘ default=None type=int
                        dest=‘val.batch_size‘ help=‘The batch size of validation.‘）

    # ***********  Params for model.  **********
    parser.add_argument（‘--model_name‘ default=None type=str
                        dest=‘network.model_name‘ help=‘The name of model.‘）
    parser.add_argument（‘--checkpoints_root‘ default=None type=str
                        dest=‘network.checkpoints_root‘ help=‘The root dir of model save path.‘）
    parser.add_argument（‘--checkpoints_name‘ default=None type=str
                        dest=‘network.checkpoints_name‘ help=‘The name of checkpoint model.‘）
    parser.add_

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2019-07-31 04:12  torchcv-master\
     文件         284  2019-07-31 04:12  torchcv-master\.gitignore
     文件       11357  2019-07-31 04:12  torchcv-master\LICENSE
     文件        7474  2019-07-31 04:12  torchcv-master\README.md
     目录           0  2019-07-31 04:12  torchcv-master\configs\
     目录           0  2019-07-31 04:12  torchcv-master\configs\cls\
     目录           0  2019-07-31 04:12  torchcv-master\configs\cls\cifar\
     文件        3035  2019-07-31 04:12  torchcv-master\configs\cls\cifar\ic_vgg19_cifar10_cls.conf
     目录           0  2019-07-31 04:12  torchcv-master\configs\cls\imagenet\
     文件        3389  2019-07-31 04:12  torchcv-master\configs\cls\imagenet\ic_shufflenetv2-50_imagenet_cls.conf
     文件        3295  2019-07-31 04:12  torchcv-master\configs\cls\imagenet\ic_shufflenetv2_imagenet_cls.conf
     目录           0  2019-07-31 04:12  torchcv-master\configs\det\
     目录           0  2019-07-31 04:12  torchcv-master\configs\det\coco\
     文件        6607  2019-07-31 04:12  torchcv-master\configs\det\coco\fasterrcnn_vgg16_coco_det.conf
     文件        6994  2019-07-31 04:12  torchcv-master\configs\det\coco\fpn_resnet50_coco_det.conf
     文件        6743  2019-07-31 04:12  torchcv-master\configs\det\coco\ssd300_vgg16_coco_det.conf
     文件        6765  2019-07-31 04:12  torchcv-master\configs\det\coco\ssd512_vgg16_coco_det.conf
     文件        6557  2019-07-31 04:12  torchcv-master\configs\det\coco\yolov3_darknet_coco_det.conf
     目录           0  2019-07-31 04:12  torchcv-master\configs\det\voc\
     文件        4546  2019-07-31 04:12  torchcv-master\configs\det\voc\fasterrcnn_vgg16_voc_det.conf
     文件        4748  2019-07-31 04:12  torchcv-master\configs\det\voc\ssd300_vgg16_voc_det.conf
     文件        4642  2019-07-31 04:12  torchcv-master\configs\det\voc\ssd512_vgg16_voc_det.conf
     文件        4342  2019-07-31 04:12  torchcv-master\configs\det\voc\yolov3_darknet_voc_det.conf
     目录           0  2019-07-31 04:12  torchcv-master\configs\gan\
     目录           0  2019-07-31 04:12  torchcv-master\configs\gan\face\
     文件        3619  2019-07-31 04:12  torchcv-master\configs\gan\face\fg_lightcnn_nir2vis_gan.conf
     文件        3612  2019-07-31 04:12  torchcv-master\configs\gan\face\it_cyclegan_nir2vis_gan.conf
     文件        3109  2019-07-31 04:12  torchcv-master\configs\gan\face\it_pix2pix_3d2vis_gan.conf
     文件        3122  2019-07-31 04:12  torchcv-master\configs\gan\face\it_pix2pix_sketch2vis_gan.conf
     目录           0  2019-07-31 04:12  torchcv-master\configs\ins\
     目录           0  2019-07-31 04:12  torchcv-master\configs\ins\coco\
............此处省略692个文件信息