摘要: 第三方库安装 albumentations 用于数据增强的库 官方文档地址 安装: 如果是conda conda install -c conda-forge imgaug conda install -c conda-forge albumentations 如果是pip pip install 阅读全文
posted @ 2021-12-20 21:59 Adam_lxd 阅读(991) 评论(0) 推荐(0) 编辑
摘要: 因为需要方法修改超参数,训练等数据,编写conf文件 import torch class config(): def __init__(self): self.BATCH_SIZE = 100 self.BASE_PATH = "你的保存数据集的文件" self.DEVICE = "cuda" i 阅读全文
posted @ 2021-12-20 21:50 Adam_lxd 阅读(325) 评论(0) 推荐(0) 编辑