随笔分类 - 深度学习
摘要:#昨天师兄给了一个组里的云GPU账号,激动的心颤动的手。。。但是Linux之前学过一点,大部分命令都忘了,用起来有点生疏,今天有时间把环境配置好了 记录一下。。。 1、有之前的miniconda3,自己开了一个虚拟的conda 环境,conda的基础命令: #查看conda版本 conda -v #
阅读全文
摘要:#为方便自己查看,比较啰嗦。。。。。 1、数据集划分(代码来自别人的分享项目中的一个文件,在项目中能跑通,单独文件能否跑通,还没试): import os import random import xml.etree.ElementTree as ET from PIL import Image i
阅读全文
摘要:1、参数解析: parser = argparse.ArgumentParser() #指定初始化权重,用于初始化训练的初始权重,可以不指定从头开始,但是用别的权重可以更快收敛 parser.add_argument('--weights', type=str, default='yolov5s.p
阅读全文