-
大小: 26.74MB文件類型: .zip金幣: 1下載: 0 次發布日期: 2023-07-15
- 語言: Python
- 標簽:
資源簡介
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個文件信息
- 上一篇:python深度學習
- 下一篇:Python 3.8.1 64位安裝包.zip
評論
共有 條評論