计算机视觉
文章平均质量分 60
真实的人
这个作者很懒,什么都没留下…
展开
-
pytorch图像分类框架搭建——利用logging记录训练日志
关于logging模块的基本使用参考Python之日志处理(logging模块),这里主要记录使用logging时踩过的坑。1. 使用pytorch的DDP进行多卡训练时如何保证只有主进程将信息发送到控制台和l日志文件中呢?logging.getLogger()方法有一个可选参数name,该参数表示将要返回的日志器的名称标识,如果不提供该参数,则其值为'root',得到的就是RootLoggerlogger的名称是一个以'.'分割的层级结构,每个'.'后面的logger都是'.'前面的logge原创 2021-08-26 15:00:30 · 1518 阅读 · 0 评论 -
pytorch图像分类框架搭建——训练流程的定义xxx_train.py
训练脚本主要包含两个函数:parse_args()用来收集参数,train()定义了整个训练流程。import argparseimport timeimport tqdmimport loggingfrom pathlib import Pathfrom datetime import datetimefrom einops import rearrangeimport sysimport torchfrom torch import nnfrom torch.utils.da.原创 2021-08-25 15:19:39 · 686 阅读 · 0 评论 -
pytorch图像分类框架搭建——项目目录,自定义文件、包导入
最近接手的项目是关于图像分类的,参考了一些优秀的代码库,对使用pytorch框架的搭建分类项目的过程做一个记录总结,同时也记录一些常用的python包的使用。原创 2021-08-24 16:21:20 · 454 阅读 · 0 评论 -
图像数据集均值与标准差的计算程序(Python多线程加速)
图像深度学习任务中,一般首先要对输入图像进行归一化,本文设计了一个可以在linux服务器上使用python多线程加速对统计数据集的均值标准差的程序原创 2021-08-10 15:49:52 · 993 阅读 · 0 评论