小猴学IT

学习记录

c++(2) Java(1) Linux(1) mmdetection(1) MySQL数据库(3) opencv(9) paddlepaddle(2) python(4) pytorch(19) ZHY项目仓库(2) 数据分析及可视化(1) 机器学习(7) 爬虫(1) 目标检测(1) 程序报错解决方案汇总(5) 自然语言处理(2) 题解(16)

/ 注册

全部文章 / 学习记录（共17篇）

python导入自定义模块失败怎么解决

1、如果导入的模块和主程序在同个目录下，直接import就行了 2、如果导入的模块是在主程序所在目录的子目录下，可以在子目录中增加一个空白的__init__.py文件，该文件使得python解释器将子目录整个也当成一个模块，然后直接通过“import 子目录.模块”导入即可。 3、如果导入的模块是在...

2021-09-30

0 1611

model.load_state_dict(state_dict, strict=False) torch.load

保存网络结构以及参数一、保存方式二、pkl、pth文件区别2.1 .pkl文件2.2 .pth文件对于pytorch保存网络参数，大家一般可以看到有 .pkl文件以及 .pth文件，对于这两者有什么区别，以及如何保存网络参数等，本文就好好讲述一下。一、保存方式首先我们知道不论是保存模型还是...

2021-08-31

0 2253

pytorch-resnet34代码实现

pytorch-resnet34 复制代码复制代码 from torch import nn import torch as t from torch.nn import functional as F class ResidualBlock(nn.Module): ### 残差单元 ...

2021-08-31

0 1388

RepVGG：使用自己的数据集进行训练、模型转换、推理

1、数据集的准备由于是分类模型，所以数据集需要准备至少两个类别，这里假设为：A、B两类。源码中是假设数据集分为两部分：train、val，分别对应训练、验证集；在train或者val数据集中，每类样本分别存放于一个独立文件夹，也即：A类的样本存放于文件夹A中，B类的样本存放于文件夹B中；因此，总的...

2021-08-31

0 628

pytorch模型转onnx模型，将onnx模型进行简化

转onnx模型 import sys sys.path.append('./data') sys.path.append('./model') from model import MobileNetV3_large import torch def convert_onnx(): d...

2021-08-26

0 1602

pip 自动添加镜像

打开anaconda prompt，输入下列指令： conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/conda config --add channels https://...

2021-08-17

0 289

pip 镜像

pypi 镜像使用帮助pypi 镜像每 5 分钟同步一次。临时使用pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package注意，simple 不能少, 是 https 而不是 http 设为默认升级 pip 到最新的版...

2021-08-06

0 327

Tensorflow安装错误之 Cannot uninstall wrapt

问题说明解决方案 pip install wrapt --ignore-installed

2021-06-05

0 367

python制作动图gif

import imageio import os path=r'C:/Users/lenovo/Pictures/test' file_list = os.listdir(path) # path = os.getcwd() frames = [] for png in file_list: ...

2021-05-18

0 372

利用yield 制作批数据集用于神经网络输入

#定义数据读取器 import cv2 import random import numpy as np # 对读入的图像数据进行预处理 def transform_img(img): # 将图片尺寸缩放道 224x224 img = cv2.resize(img, (224, 2...

2021-05-07

0 418