argparse标准库在pytorch-grad-cam实战中的使用分析是怎样的
这篇文章给大家介绍argparse标准库在pytorch-grad-cam实战中的使用分析是怎样的,内容非常详细,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助。
在长顺等地区,都构建了全面的区域性战略布局,加强发展的系统性、市场前瞻性、产品创新能力,以专注、极致的服务理念,为客户提供成都做网站、网站建设 网站设计制作按需策划设计,公司网站建设,企业网站建设,品牌网站制作,全网整合营销推广,外贸网站建设,长顺网站建设费用合理。
代码展示:
import torchimport argparsedef get_args():parser = argparse.ArgumentParser()parser.add_argument('--use-cuda', action='store_true', default=False,help='Use NVIDIA GPU acceleration')parser.add_argument('--image-path', type=str, default='./examples/both.png',help='Input image path')args = parser.parse_args()args.use_cuda = args.use_cuda and torch.cuda.is_available()if args.use_cuda:print("Using GPU for acceleration")else:print("Using CPU for computation")return args args = get_args()print("打印读取的信息".center(50,'-'))print('args.use_cuda:', args.use_cuda)print('args.image_path:', args.image_path)# python testArgs.py -h# python testArgs.py# python testArgs.py --use-cuda# python testArgs.py --use-cuda --image-path 1.jpg # python testArgs.py --image-path 1.jpg --use-cuda
控制台下执行结果展示:
Windows PowerShell 版权所有 (C) Microsoft Corporation。保留所有权利。 尝试新的跨平台 PowerShell https://aka.ms/pscore6 加载个人及系统配置文件用了 943 毫秒。(base) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq> conda activate ssd(ssd) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq> python testArgs.py -h usage: testArgs.py [-h] [--use-cuda] [--image-path IMAGE_PATH]optional arguments: -h, --help show this help message and exit --use-cuda Use NVIDIA GPU acceleration --image-path IMAGE_PATH(ssd) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq> python testArgs.pyUsing CPU for computation---------------------打印读取的信息----------------------args.use_cuda: False(ssd) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq> python testArgs.py --use-cudaUsing GPU for acceleration---------------------打印读取的信息----------------------args.use_cuda: True args.image_path: ./examples/both.png(ssd) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq> python testArgs.py --image-path 1.jpgUsing CPU for computation---------------------打印读取的信息----------------------args.use_cuda: False args.image_path: 1.jpg(ssd) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq> python testArgs.py --use-cuda --image-path 1.jpgUsing GPU for acceleration---------------------打印读取的信息----------------------args.use_cuda: True args.image_path: 1.jpg(ssd) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq> python testArgs.py --image-path 1.jpg --use-cudaUsing GPU for acceleration---------------------打印读取的信息----------------------args.use_cuda: True args.image_path: 1.jpg(ssd) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq>
关于argparse标准库在pytorch-grad-cam实战中的使用分析是怎样的就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。
新闻名称:argparse标准库在pytorch-grad-cam实战中的使用分析是怎样的
本文来源:http://hbruida.cn/article/iehjss.html