![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
weixin_36049506
这个作者很懒,什么都没留下…
展开
-
AttributeError: 'Tensor' object has no attribute 'argsort'
image_pred = image_pred[(-score).argsort()] yolov3程序调试时出错:pytorch的tensor没有argsort方法 不希望转成numpy再转回tensor,查阅手册发现 pytorch的sort同时返回了sorted和indices,代码修改为: srtd, indices = torch.sort(-score) image_pred = ...原创 2019-12-08 01:37:43 · 2348 阅读 · 0 评论 -
pascal转coco 按比例划分数据
# -*- coding:utf-8 -*- # !/usr/bin/env python import argparse import json import matplotlib.pyplot as plt import skimage.io as io import sys sys.path.remove('/opt/ros/kinetic/lib/python2.7/dist-pack...原创 2019-11-12 15:18:05 · 160 阅读 · 0 评论 -
pytorch载入预训练模型后,训练指定层
1、有了已经训练好的模型参数,对这个模型的某些层做了改变,如何利用这些训练好的模型参数继续训练: pretrained_params = torch.load('Pretrained_Model') model = The_New_Model(xxx) model.load_state_dict(pretrained_params.state_dict(), strict=False) stric...原创 2019-04-25 19:08:30 · 2014 阅读 · 0 评论 -
入坑pytorch之pytorch基础
文章目录PyTorch 动态计算图Pytorch张量第一个网络mnist网络初始化和前向传播dataloader/dataprivder优化器和损失函数训练过程完整代码 PyTorch 动态计算图 在讨论 PyTorch 的各个组件前,我们需要了解它的工作流。PyTorch 使用一种称之为 imperative / eager 的范式,即每一行代码都要求构建一个图,作为定义完整计算图的一个部分。即...原创 2019-04-22 20:08:51 · 266 阅读 · 0 评论 -
Module.train Module.eval 方法和 Module.training属性的关系
Module.train()、Module.eval() 方法和 Module.training属性的关系 print(net.training, net.submodel1.training) net.train() # 将本层及子层的training设定为True net.eval() # 将本层及子层的training设定为False net.training = True # 注意,对mo...原创 2019-05-22 10:44:46 · 1004 阅读 · 0 评论 -
python工厂模式
在faster-rcnn的数据集读取部分,有一个factory.py文件(为了便于阅读删除了一些代码,比如image_net数据集的处理,但逻辑没变): __sets = {} from lib.datasets.pascal_voc import pascal_voc from lib.datasets.coco import coco import numpy as np # Set up...原创 2019-07-04 16:09:32 · 101 阅读 · 0 评论