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