Python-图像分类目标检测姿态估计分割的Pytorch实现
标签:
•
文件类型: .zip
•
文件大小: 26.74MB
•
下载次数: 1
•
Repo for most of CV problems, such as image classification, object detection, pose estimation, segmentation, and so on.
代码片段和文件信息
属性 大小 日期 时间 名称
----------- --------- ---------- ----- ----
目录 0 2019-07-31 04:12 torchcv-master
文件 284 2019-07-31 04:12 torchcv-master.gitignore
文件 11357 2019-07-31 04:12 torchcv-masterLICENSE
文件 7474 2019-07-31 04:12 torchcv-masterREADME.md
目录 0 2019-07-31 04:12 torchcv-masterconfigs
目录 0 2019-07-31 04:12 torchcv-masterconfigscls
目录 0 2019-07-31 04:12 torchcv-masterconfigsclscifar
文件 3035 2019-07-31 04:12 torchcv-masterconfigsclscifaric_vgg19_cifar10_cls.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsclsimagenet
文件 3389 2019-07-31 04:12 torchcv-masterconfigsclsimagenetic_shufflenetv2-50_imagenet_cls.conf
文件 3295 2019-07-31 04:12 torchcv-masterconfigsclsimagenetic_shufflenetv2_imagenet_cls.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsdet
目录 0 2019-07-31 04:12 torchcv-masterconfigsdetcoco
文件 6607 2019-07-31 04:12 torchcv-masterconfigsdetcocofasterrcnn_vgg16_coco_det.conf
文件 6994 2019-07-31 04:12 torchcv-masterconfigsdetcocofpn_resnet50_coco_det.conf
文件 6743 2019-07-31 04:12 torchcv-masterconfigsdetcocossd300_vgg16_coco_det.conf
文件 6765 2019-07-31 04:12 torchcv-masterconfigsdetcocossd512_vgg16_coco_det.conf
文件 6557 2019-07-31 04:12 torchcv-masterconfigsdetcocoyolov3_darknet_coco_det.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsdetvoc
文件 4546 2019-07-31 04:12 torchcv-masterconfigsdetvocfasterrcnn_vgg16_voc_det.conf
文件 4748 2019-07-31 04:12 torchcv-masterconfigsdetvocssd300_vgg16_voc_det.conf
文件 4642 2019-07-31 04:12 torchcv-masterconfigsdetvocssd512_vgg16_voc_det.conf
文件 4342 2019-07-31 04:12 torchcv-masterconfigsdetvocyolov3_darknet_voc_det.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsgan
目录 0 2019-07-31 04:12 torchcv-masterconfigsganface
文件 3619 2019-07-31 04:12 torchcv-masterconfigsganfacefg_lightcnn_nir2vis_gan.conf
文件 3612 2019-07-31 04:12 torchcv-masterconfigsganfaceit_cyclegan_nir2vis_gan.conf
文件 3109 2019-07-31 04:12 torchcv-masterconfigsganfaceit_pix2pix_3d2vis_gan.conf
文件 3122 2019-07-31 04:12 torchcv-masterconfigsganfaceit_pix2pix_sketch2vis_gan.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsins
目录 0 2019-07-31 04:12 torchcv-masterconfigsinscoco
............此处省略692个文件信息
#!/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-masterLICENSE
文件 7474 2019-07-31 04:12 torchcv-masterREADME.md
目录 0 2019-07-31 04:12 torchcv-masterconfigs
目录 0 2019-07-31 04:12 torchcv-masterconfigscls
目录 0 2019-07-31 04:12 torchcv-masterconfigsclscifar
文件 3035 2019-07-31 04:12 torchcv-masterconfigsclscifaric_vgg19_cifar10_cls.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsclsimagenet
文件 3389 2019-07-31 04:12 torchcv-masterconfigsclsimagenetic_shufflenetv2-50_imagenet_cls.conf
文件 3295 2019-07-31 04:12 torchcv-masterconfigsclsimagenetic_shufflenetv2_imagenet_cls.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsdet
目录 0 2019-07-31 04:12 torchcv-masterconfigsdetcoco
文件 6607 2019-07-31 04:12 torchcv-masterconfigsdetcocofasterrcnn_vgg16_coco_det.conf
文件 6994 2019-07-31 04:12 torchcv-masterconfigsdetcocofpn_resnet50_coco_det.conf
文件 6743 2019-07-31 04:12 torchcv-masterconfigsdetcocossd300_vgg16_coco_det.conf
文件 6765 2019-07-31 04:12 torchcv-masterconfigsdetcocossd512_vgg16_coco_det.conf
文件 6557 2019-07-31 04:12 torchcv-masterconfigsdetcocoyolov3_darknet_coco_det.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsdetvoc
文件 4546 2019-07-31 04:12 torchcv-masterconfigsdetvocfasterrcnn_vgg16_voc_det.conf
文件 4748 2019-07-31 04:12 torchcv-masterconfigsdetvocssd300_vgg16_voc_det.conf
文件 4642 2019-07-31 04:12 torchcv-masterconfigsdetvocssd512_vgg16_voc_det.conf
文件 4342 2019-07-31 04:12 torchcv-masterconfigsdetvocyolov3_darknet_voc_det.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsgan
目录 0 2019-07-31 04:12 torchcv-masterconfigsganface
文件 3619 2019-07-31 04:12 torchcv-masterconfigsganfacefg_lightcnn_nir2vis_gan.conf
文件 3612 2019-07-31 04:12 torchcv-masterconfigsganfaceit_cyclegan_nir2vis_gan.conf
文件 3109 2019-07-31 04:12 torchcv-masterconfigsganfaceit_pix2pix_3d2vis_gan.conf
文件 3122 2019-07-31 04:12 torchcv-masterconfigsganfaceit_pix2pix_sketch2vis_gan.conf
目录 0 2019-07-31 04:12 torchcv-masterconfigsins
目录 0 2019-07-31 04:12 torchcv-masterconfigsinscoco
............此处省略692个文件信息
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件举报,一经查实,本站将立刻删除。
评论列表(条)