![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
xiaoju233
Talk is cheap, show me the code.
展开
-
pytorch手把手使用FasterRCNN进行目标检测
手把手使用FasterRCNN进行目标检测使用pytorch自带的预训练FasterRCNN目标检测模型进行目标检测,并把检测到的结果打印到原图片上。由于pytorch自带的FasterRCNN是使用COCO数据集进行预训练得到的,所以我们需要使用COCO的标签来匹配模型输出的结果。详细的注释都在代码里面了。import torchimport torchvisionfrom torchvision.models.detection.roi_heads import fastrcnn_loss原创 2021-10-06 20:49:13 · 5699 阅读 · 6 评论 -
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same出现这个问题是因为输入模型的tensor加载在cpu中,而模型却加载在cuda上。解决方法:将输入tensor加载到cuda中,或者把模型加载到cpudevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")mod原创 2021-10-03 11:56:57 · 1673 阅读 · 1 评论 -
pytorch读取与保存网络模型
pytorch读取与保存网络模型方法一:只保存网络模型的参数,而不保存网络的结构保存:torch.save(model.state_dict(), 'model_params.pth')读取:model.load_state_dict(torch.load('model_params.pth'))方法二:同时保持网络的结构和参数保存:torch.save(model, 'model.pth')读取:model = torch.load('model.pth')...原创 2020-06-26 13:36:12 · 1020 阅读 · 0 评论 -
pytorch使用指定的GPU
###pytorch使用指定的GPU在实践中,往往会有一些显卡被其它任务占用,这时我们需要指定使用空闲的GPU进行工作。通过环境变量指定GPUexport CUDA_VISIBLE_DEVICES=0,1,2,3在程序开头设置os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0,1,2,3’在命令行执行的命令前添加CUDA_VISIBLE_DEVICES=0,1,2,3,4使用torch.cuda接口torch.cuda.set_device(0)亲测貌似不行原创 2020-06-13 23:48:19 · 203 阅读 · 0 评论