摘要: def download_pretrained_vae(overwrite=False): download_path = "pretrained_models/vae/kl16.ckpt" if not os.path.exists(download_path) or overwrite: hea 阅读全文
posted @ 2024-09-20 12:09 Undefined443 阅读(207) 评论(0) 推荐(0)
摘要: 这里记录一些我用过的数据集 可以从 TensorFlow Datasets Catalog 找到几乎所有数据集的 source。 torch.utils.data.Dataset 提供了常见数据集的读取类。 ImageNet 适用任务: 图像分类 目标定位 目标检测 经常使用的是 ImageNet 阅读全文
posted @ 2024-09-20 09:56 Undefined443 阅读(174) 评论(0) 推荐(0)