![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
python学习
zyoung17
这个作者很懒,什么都没留下…
展开
-
numpy寻找list或者数组某一维相同元素的索引
问题描述在做检测任务时,想要寻找相同语义类别的所有框的索引。描述框的数组为K*4,其中最后一个维度代表语义类别,目的是找出相同语义类别(即最后一维各种相同元素)的索引。解决方法主要使用一个for循环以及np.argwhere()函数来解决。import numpy as nplist = [1, 2, 3, 4, 2, 2, 3, 1]for i in np.unique(list): idx = np.argwhere(list == i) print('index fo原创 2020-11-06 21:15:49 · 4602 阅读 · 2 评论 -
python[数据处理]把txt/excel的文件夹名对应的内容复制到一个新的文件夹
问题描述整理数据集,一个大的文件夹A下面有1-100个子文件夹,每个子文件夹里面的图片都属于同一类,另外有一个或多个excel表或者txt文件(每个excel或者txt代表一类)分别注明这类样本对应的文件夹名。目的是把所有同一个类别的图片放到同一个文件夹内。写了一个以下的脚本来实现。实现代码import osimport shutilwith open('./A类.txt', 'r') as f: dic_name_list = f.read().splitlines()# print原创 2020-09-23 19:43:20 · 553 阅读 · 0 评论 -
PyTorch损失函数之交叉熵损失函数nn.CrossEntropyLoss()
PyTorch之nn.CrossEntropyLoss()nn.CrossEntropyLoss()是nn.logSoftmax()和nn.NLLLoss()的整合,可以直接使用它来替换网络中的这两个操作,这个函数可以用于多分类问题。具体的计算过程可以参考官网的公式或者一下这个链接。https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html#torch.nn.CrossEntropyLosshttps://blog.原创 2020-09-06 13:41:26 · 27953 阅读 · 9 评论 -
安装Pytorch
在Windows环境下安装PyTorch安装anaconda,参照链接Anaconda入门按照路径Start | Anaconda3 | Anaconda Prompt打开windows下的终端框。将pytorch安装在对应的虚拟环境中便于管理,使用如下命令创建虚拟环境并进入环境。也可以使用Anaconda软件的图形化界面来创建虚拟环境。conda create -n torch py...原创 2019-12-09 19:54:06 · 472 阅读 · 0 评论 -
PyCharm建立工程(含选Project Interpreters)
最开始的问题是安装了pytorch框架,在终端里面检测一直是显示安装成功的,但在pycharm里面import torch一直报错。基本更换了所有的解释器,折腾了一天发现都不行。最后找到是工程建立的问题。写个帖子规范一下自己建立PyCharm工程。File --New Project --输入工程名(点下面一行那个小三角)–再选择Existing interpreter找到自己的解释器路径...原创 2019-10-15 11:14:36 · 1099 阅读 · 0 评论 -
python学习笔记
在python学习过程中使用矩阵相乘函数发现无法实现向量和矩阵相乘。报错ValueError: Shape must be rank 2 but is rank 1 for ‘MatMul’ (op: ‘MatMul’) with input shapes: [2], [2,1].查找原因发现相乘的两个矩阵格式不对。import tensorflow as tfx = tf.constan...原创 2019-10-09 16:36:47 · 83 阅读 · 0 评论 -
一种比较好的描述数组段的方法
说明:比如我要取用长度为32的一维数组,每次取8个(就是分别[0,7] [8,15] [16,23] [24,31]),记下来提醒一下自己。代码可以这样实现:for i in range(0,4) #(0,4是前开后闭)start = (i*BATCH_SIZE) % 32end = start + BATCH_SIZE #此处的BATCH_SIZE应该为7然后[start , end...原创 2019-10-09 17:41:23 · 146 阅读 · 0 评论