摘要:
虽然 images,labels = next(iter(test_data)) 可以每次1batch获取,但是超级慢,不推荐。 import torch import platform if platform.system() == 'Windows': import matplotlib mat 阅读全文
摘要:
以Halcon为例,安装的软件是不带图标的,在网上下载个,命名为halcon.jpg放到了安装路径里 桌面处——鼠标右键——打开终端,我已经安装了VSCode(习惯用它来编辑文本) code halcon.desktop 然后粘贴如下内容,保存后关闭VSCode。具体含义:logo下边的名字、可执行 阅读全文
摘要:
推荐torchinfo,官网地址 TylerYep/torchinfo: View model summaries in PyTorch! torchsummary、torch-summary都用过,还是torchinfo更准,尤其是显示yolo模型的时候。 展示形式如下: 对于一般模型,如resn 阅读全文
摘要:
1、Simd库——CPU指令集加速 算子 Simd Library Documentation. 部分算子截图: 2、VPI库——CPU、GPU(CUDA)加速 算子 VPI - Vision Programming Interface: Algorithms 部分算子截图: 3、CV-CUDA库 阅读全文
摘要:
1、Softmax(dim=1)的含义 num_classes = 2 #几分类问题就写几 resnet50 = models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V2) fc_inputs = resnet50.fc.in_fea 阅读全文
摘要:
分为三部分:常用软件、训练用软件、部署用软件 训练都是基于python的,所以安装在虚拟环境里 部署是基于c++的,所以安装在本地 1、常用软件安装 【图像标注工具labelImg】 官方地址 https://github.com/HumanSignal/labelImg ,其中又新出来个工具 Im 阅读全文
摘要:
1、RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same 如上错误是在 torch.onnx.export 时出现的 解决:x造成的,后边加上. 阅读全文
摘要:
以500*332的图为例 import torchvision.transforms as transforms from PIL import Image img = Image.open('D:/cat.jpg') #缩放成高宽都是256的正方形(小变大的话,是插值不会填充黑边) resize 阅读全文
摘要:
1、样本集下载与划分 下载,https://download.microsoft.com/download/3/E/1/3E1C3F21-ECDB-4869-8368-6DEBA77B919F/kagglecatsanddogs_5340.zip 划分,将Cat和Dog文件夹里的图片按3:1:1复制 阅读全文
摘要:
注意numpy版本不要使用2.x,否则提醒 AttributeError: `np.string_` was removed in the NumPy 2.0 release. Use `np.bytes_` instead. 安装1.x版本,查看版本 conda search numpy 安装1. 阅读全文