pytorch
yuyijie_1995
半路出家菜鸡程序员一枚
展开
-
fasterrcnn使用IOU损失
推理发现 利用fasterrcnn得到的dx,dy,dw,dh 和target中的gt的dx,dy,dw,dh就可以求得IOU完全不需要变化转换代码def bbox_transform(deltas, weights): wx, wy, ww, wh = weights dx = deltas[:, 0::4] / wx dy = deltas[:, 1::4] ...原创 2019-12-21 21:49:48 · 1261 阅读 · 4 评论 -
mmdetection 修改预训练模型权重类别数
修改预训练权重类别数import osimport torchimport argparsedef init_args(): parser = argparse.ArgumentParser() parser.add_argument("--org_path", type=str, help="the path of pretrained model") parse...原创 2019-12-11 14:10:48 · 2433 阅读 · 6 评论 -
Simple-fasterrcnn源码学习笔记(5)
这篇是最后一篇 讲述train.py和trainer.py的一些代码细节文章目录train.pytrainer.py不熟悉的python,pytorch,numpy操作fire 模块data_.DataLoaderself.faster_rcnn.state_dict()t.save(save_dict,save_path)self.faster_rcnn.load_state_dict(sta...原创 2019-11-26 22:29:37 · 494 阅读 · 0 评论 -
Simple-fasterrcnn源码学习笔记(4)
这里主要是utils下的部分文件的补充注释文章目录config.pyarray_tool.pyeval_tool.pyvis_tool.py不熟悉的numpy,pytorch,python操作data.detach()data.numpy()t.from_numpy(data)data.reshape(1)[0]data.item()from collections import default...原创 2019-11-26 21:49:54 · 475 阅读 · 0 评论 -
Simple-fasterrcnn源码学习笔记(3)
这部分主要讲述model部分的代码,是最关键的部分。文章目录整个模型代码跳转过程整个模型代码跳转过程首先描述整个训练的流程,再根据流程逐一解释代码1.train.py 中会先定义faster_rcnn=faster_rcnn_vgg16()#定义好模型这里将调用fasterrcnnvgg16.py文件中的class FasterRCNNVGG16(FasterRCNN)进行初始化工作...原创 2019-11-26 10:53:36 · 649 阅读 · 0 评论 -
Simple-fasterrcnn源码学习笔记(2)
文章目录bbox_tools_enumerate_shift_anchor.pycreator_tools.pynumpy,pytorch,python不熟的操作eps=np.finfo(height.dtype).eps 防止分母为0的操作np.vstack((dy,dx,dh,dw)).transpose()dy=loc[:,0::4] #这里为什么不直接0src_height[:,np.ne...原创 2019-11-23 18:16:25 · 756 阅读 · 0 评论 -
Simple-fasterrcnn源码学习笔记 (1)
文章目录voc_dataset.pydataset.pytsf数据预处理过程util.py_config.py这部分代码不熟悉的numpy,python,pytorch操作小结bbox=np.stack(bbox).astype(np.float32)img[::-1,:,:]numpy,tensor互转对numpy数据增加一个新的轴numpy数据维度互换位置从几个选项中选一个阅读该博客 ht...原创 2019-11-23 15:15:19 · 532 阅读 · 0 评论 -
Pytorch学习笔记
文章目录数据操作index_select(input,dim,index)masked_select(input,mask)nonzero(input)gather(input,dim,index)view() 改变张量的形状item()内存开销Tensor,Numpy互转Tensor 放到GPU上自动求梯度requires_grad梯度实战线性回归数据输入定义模型nn.Modulenet.par...原创 2019-11-14 18:39:01 · 391 阅读 · 0 评论 -
Pytorch中使用指定的GPU
1.直接终端中设定:CUDA_VISIBLE_DEVICES=1 python my_script.py2.python代码中设定:import osos.environ["CUDA_VISIBLE_DEVICES"]="2"3.使用函数 set_deviceimport torchtorch.cuda.set_device(id)...原创 2019-11-11 10:34:59 · 189 阅读 · 0 评论