train.py

MMdetection之train.py源码详解

目录一、tools/train.py二、源码详解三、核心函数详解（一）build_detector（mmdet/models/builder.py）（二）build_dataset（mmdet/datasets
小刺猬69·2022-12-17 14:19

yolov5——断点训练/继续训练【解决方法、使用教程】

放心，在yolov5中给我们提供现成的参数--resume解决方法先将train.py中的--resume的default改为True，表示默认是打开状态接着在runs/train/exp*/weights
Mr_浮熙·2022-12-16 23:14

U版YOLOv5学习笔记之训练参数

源码链接：https://github.com/ultralytics/yolov5/blob/master/train.py使用字典存储模型的超参数#Hyperparametershyp={'lr0'
Jekk_cheng·2022-12-16 23:43

YOLOV5训练过程中想减少或者增加epoch

版本YOLOV5v6.1首先要Ctrl+C中止训练，然后到train.py，修改parse_opt函数中的‘–resume’将default=False改为default=True，在485行左右，如下图
qingfeng253·2022-12-16 23:12

python 导入同级目录文件、上级目录文件以及下级目录数据集和模块包

一.pythonimport上级目录：1.1第一种方法：有时候我们可能需要import另一个路径下的python文件，例如下面这个目录结构，我们想要在_train.py里import在networks目录下的
cv_lhp·2022-12-16 17:20

RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

问题现象：raceback(mostrecentcalllast): File"train.py",line100,in  main(opt) File"train.py",line71,inmain…
flashlau·2022-12-16 00:37

Detectron2注册自己的COCO数据集

Detectron2注册自己的COCO数据集1.在train.py添加CLASS_NAMES=['background','A220','A330','A320/321','Boeing737-800'
SSSLaker·2022-12-15 20:18

OSError: libmkl_intel_lp64.so: cannot open shared object file: No such file or directory

Traceback(mostrecentcalllast):File"train.py",line11,inimporttorch.distributedasdistFile"/home/gbstack
gbstack08·2022-12-15 14:39

pyTorch入门（三）——GoogleNet和ResNet训练

——《微卡智享》本文长度为2748字，预计阅读8分钟前言这是Minist训练的第三篇了，本篇主要是把GoogleNet和ResNet的模型写出来做一个测试，再就是train.py里面代码加入了图例显示。
Vaccae·2022-12-15 05:21

使用YOLOX训练自己的数据集

文章目录环境准备数据集准备安装YOLOX修改配置文件修改检测类别修改训练参数修改训练数据集路径修改Voc.py文件中Annotations文件的读取格式训练修改train.py查看实时训练情况检测环境准备需要
SmallSweetWang·2022-12-15 01:50

yolov5中的best.pt

YOLOv5Documentation所以只好一行行看train.py的源码，在下图位置看到了best.pt的更新ctrl加鼠标左键点击fitness，来到了utils里的metrics类这里就一
EXYlkx·2022-12-14 23:17

yolov5 继续训练

直接在train.py设置resume参数为True即可parser.add_argument('--resume',n
weixin_44057060·2022-12-14 18:55

一步真实解决：AssertionError：‘ERROR ：--resume checkpoint does not exist‘

我也遇到了这，最开始是直接把【train.py】里面的这个参数【--resume】的default设置成【True】,但是出现了下面的问题。
蓝胖胖▸·2022-12-14 18:24

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace

Traceback(mostrecentcalllast):File"train.py",line182,intrain(config)File"train.py",line120,intrainloss.backward
ZQ_ZHU·2022-12-14 13:54

RuntimeError: a leaf Variable that requires grad has been used in an in-place operation

Traceback(mostrecentcalllast):File“E:\迅雷下载\向量\000、代码+数据+课件\YOLO5\yolov5-master\train.py”,line466,intrain
爱吃肉爱睡觉的Esther·2022-12-14 13:50

Swin Transformer进行语义分割，Ubuntu系统，解决RuntimeError: one of the variables needed for gradient computation

1、问题描述最近在用SwinTransformer做语义分割的任务，基本的配置环境参考的是GitHub的基础安装步骤，主要的是环境的配置问题，mmcv的配置要求，不允许高于1.4.0，所以，在train.py
hacker_NO_007·2022-12-13 17:48

基于PaddleOCR的DBNet多分类文本检测网络

目录目的模型网络结构对比代码实现1、数据集格式2、配置文件调整3、数据预处理4、模型代码调整5、添加多分类loss6、修改db_postprocess.py7、修改train.py、eval.py、infer_det.py
番茄小能手·2022-12-12 14:38

yolov5 EOFError: Ran out of input

删除重新训练会自动生成，不删除可能因为错误数据导致加载不了数据集Traceback(mostrecentcalllast):File"D:/JaskProbject/Recaptcha/yolov5_word/train.py
_JackSparrow·2022-12-12 13:33

MMDetection框架入门教程（三）：配置文件详细解析

  在上一篇博客中提到，MMDetection搭建训练算法只需要3个步骤：1)准备数据集2)编写配置文件3)执行train.py文件开始训练。
Maples丶丶·2022-12-12 13:01

faster rcnn中train.py

这是一个简单的solver包装类，主要是为了实现自己的snapshot，值得一提的地方不是太多，主要是为了读者从头到尾的训练过程理解更加连贯，所以为此文单独开一节源码分析。classSolverWrapper(object):"""AsimplewrapperaroundCaffe'ssolver.Thiswrappergivesuscontroloverhesnapshottingprocess
qq小小布丁·2022-12-12 10:35

cv2.error: Caught error in DataLoader worker process 0.

问题描述在复现Tianxiaomo/pytorch-YOLOv4的时候出现如下错误解决方案在train.py中搜索train_loader，将num_workers改成0
canpian7·2022-12-11 12:27

物体检测框架之mmdetection源码阅读（1）

通过readme不难发现单卡训练是以tools/train.py作为程序的入口。首先贴出train.py的源代码。
liqiming100·2022-12-11 11:08

有效解决译运行报错undefined symbol: _ZN10tensorflow12OpDefBuilder4AttrESs

在编译makefile文件生成.so文件后，发现运行train.py时，出现报错:undefinedsymbol:_ZN10tensorflow12OpDefBuilder4AttrESs。
weixin_44119674·2022-12-11 10:33

mmrotate自己训练修改参数

1.tools/train.py修改训练所需的模型和训练获得权重的目标文件夹（文件夹会自动创建）parser.add_argument('--config',defualt='/home/zhh/download
青椒肉丝_·2022-12-10 16:04

AssertionError: train: No labels in D:\******\data；Can not train without labels.

github.com/ultralytics/yolov5YOLOv5自定义数据集_汤俊杰2131312的博客-CSDN博客二、解决思路最近在做yolo算法的目标检测任务时，把自己做的数据集设置完毕后，运行train.py
键盘即钢琴·2022-12-10 08:31

train和eval在训练集上表现差距很大

2.为避免代码造成的问题，于是将train.py基础上去掉后向传播和模型保存部分，结果与上述一致。3.根据网上的一些办法，使用以下函数冻结bn
MlllllllllJ.·2022-12-09 19:41

PyTorch量化报错后端不匹配

PyTorch-1.7.1错误描述：使用PyTorchQuantization包进行量化感知训练（QAT）时，最后一步convert报错：Traceback(mostrecentcalllast):File"train.py
YulongYang1896·2022-12-09 11:28

训练 train.py

1.随机种子点通过设置随机种子点保证实验结果的可复现性cudnn.deterministictorch.backends.cudnn.deterministic=True的话，每次返回的卷积算法将是确定的，即默认算法cudnn.benchmark设置torch.backends.cudnn.benchmark=True将会让程序在开始时花费一点额外时间，为整个网络的每个卷积层搜索最适合它的卷积实现
yunxi1008·2022-12-09 04:48

mmsegmentation之tools/train.py文件解析(部分，持续更新)

#Copyright(c)OpenMMLab.Allrightsreserved.importargparseimportcopyimportosimportos.pathasospimporttimeimportwarningsimportmmcvimporttorchimporttorch.distributedasdistfrommmcv.cnn.utilsimportrevert_sync
kira_Y·2022-12-09 04:44

SRGAN代码逐行精读并注释——CSDN第一人

之前在CSDN查找过有关SRGAN代码的解读文章，奈何没有一篇文章做到了对SRGAN的完全解读，所以本人打算打破这尴尬的处境，对SRGAN的data_utils.py，model.py，loss.py，train.py
雷神-尼卡·2022-12-08 22:36

无sudo权限解决ImportError: /lib64/libstdc++.so.6: version `GLIBCXX_3.4.21‘ not found

在服务器上使用pytorch导入tensorboard时，出现以下错误：Traceback(mostrecentcalllast):File"train.py",line3,infromtorch.utils.tensorboardimportSummaryWriterFile
RRRUAAA·2022-12-08 12:12

UNet复现及环境配置（含数据集）

运行直接运行train.py文件即
.小简.·2022-12-08 09:15

pytorch基础

pytorch基础1.1损失函数与反向传播1.2优化器1.3VGG16模型的使用与修改1.4完整网络模型的训练套路简单汇总1.4.1网络模型的保存与读取1.4.2完整模型套路创建网络模型（model.py要确保与train.py
HSR CatcousCherishes·2022-12-08 03:35

mac ModuleNotFoundError: No module named ‘mmcv‘

s2anet_r50_fpn_1x_dota_le135.pyTraceback(mostrecentcalllast):File"/Users/shenhaibb/Downloads/mmrotate/tools/train.py
shenhaibb_·2022-12-07 20:28

训练DPT:由测试test到训练train图像的一个epochs的optimize.zero_grad() loss.backward() optimizer.step()

不知道大家有没有这样的感受，很多研究型论文通常会给出他们的test.py文件，但是其train.py文件往往是空白的，这时候感觉文章的test确实很nice，就想去探究其更原始、最优参数，训练出的参数过程
MengYa_DreamZ·2022-12-07 19:20

在kaggle上用Yolov5训练口罩模型并部署到android端

文章目录前言一、使用步骤（一）下载Yolov5源码（二）配置Yolov5所需的库（三）修改源码1.修改输出文件的保存路径2.添加mask.yaml3.修改models4.配置train.py（四）在Kaggle
Yellow Small Tiger·2022-12-07 18:56

GAnet A Keypoint-based Global Association Network for Lane Detection 代码调试与阅读笔记

首先从train.py文件开始阅读trian.pyargs=parse_args(）调用parse_args()函数为解析命令行参数进行配置defparse_args():parser=argparse.ArgumentParser
举焰·2022-12-07 17:20

基于YOLO的谱面识别与检索【Python环境实现】

.YOLO模型结构学习与介绍1.模型结构下载2.模型结构介绍二．YOLO模型包训练输出结果调试三．YOLO模型乐谱识图训练过程1.训练集路径设置2.修改配置文件3.数据集标注与导入4.硬件环境5.修改train.py
物质波波波·2022-12-06 17:00

在tf1.15的环境中运行github上slot-gated代码，出现rnn_cell_impl中没有rnn_cell_impl属性的问题。已解决

Traceback(mostrecentcalllast):File"G:\python_project\Slot-Gated\SlotGated-SLU\train.py",line188,intraining_outputs
student_low·2022-12-05 02:59

【目标检测】YOLO v5 吸烟行为识别检测

吸烟行为目标检测模型：计算机配置、制作数据集、训练、结果分析和使用前言相关连接（look评论）一、计算机配置pytorch安装-GPU版本pycocotools的安装二、YOLOv5下载三、模型使用获取数据集更改train.py
young_py·2022-12-03 23:35

YOLOV3训练自己的数据集报错 Expected size 85 but got size 6 for tensor number 2 in the list.

学习记录报错：Traceback(mostrecentcalllast):File"train.py",line256,inrun()File"train.py",line234,inrunmetrics_output
哒哒嗒·2022-12-03 20:07

ViT-pytorch 复现

代码地址：https://github.com/jeonsworld/ViT-pytorch1.报错Traceback(mostrecentcalllast):File"train.py",line16
深度学习浅行者·2022-12-03 17:23

yolov5 继续训练

轮，发现模型未收敛，想再继续训练200.采用博主博客中方法2，成功跑起来了感谢博主：yolov5继续训练_weixin_44057060的博客-CSDN博客_yolov5继续训练具体做法是：不是直接在train.py
m0_46093829·2022-12-03 14:59

RetinaFace-Pytorch

1.前言我先是想跑原作者biubug6的源码,但是没有跑动.原因是爆显存,debug模式下发现train.py被反复执行直到显存爆满停止.debug失败qwq.贴一下源码地址原作者RetinafaceGithub
每次都来学点儿·2022-12-03 13:27

Glow-pytorch复现github项目

本仓库中的代码主要在model.py和train.py中，结构、逻辑
zzfive·2022-12-03 11:24

【超分辨率】【深度学习】SRGAN pytorch代码（附详细注释和数据集链接）

SRGAN前言1数据集预处理2train.py3test_image.py4loss.py5model.py6data_utils.py前言主要改进部分：断点恢复（参考train.py的75行）注释部分代码
zhanjuex·2022-12-03 11:24

YOLOv7训练过程中遇到的问题及解决措施

训练过程中遇到的问题数据集上传到tmp权重文件不能和数据集一起上传到tmp中，要上传到对应映射的文件夹下，例如，找到最新的best.pt文件，将最新的best.pt文件下载到本地，然后从本地上传到对应映射的文件夹下，train.py
卿云阁·2022-12-03 10:50

【深度学习】搭建类似LeNet-5网络识别Fashion-MNIST数据集(衣服，鞋子等)

文章目录前言一、基础知识二、train.py代码模块讲解2.0.调库2.1数据载入2.2网络搭建2.3网络训练2.4train.py总体代码三、test.py代码模块讲解3.1第一种测试方法3.2第二种测试方法
今天一定要洛必达·2022-12-02 19:59

为什么GitHub的deepwalk和OpenNE-pytorch里都没有类似train.py和main.py的文件，那要怎么训练？

https://github.com/phanein/deepwalk或许，OpenNE-pytorch里有，是supervied_nodes_classfition.py这样的？
hamimelon2020·2022-12-02 14:19

手把手教你使用LabVIEW OpenCV DNN实现手写数字识别（含源码）

.OpenCVDNN简介2.LabVIEW中DNN模块函数二、TensorFlowpb文件的生成和调用1.TensorFlow2Keras模型（mnist）2.使用Keras搭建cnn训练mnist（train.py
virobotics·2022-12-02 13:38
上一页 4 5 6 7 8 9 10 11 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他