文件名称：DeepLabv3FineTuning:在您的PyTorch中为自己的细分任务微调DeepLabv3细分网络的教程
文件大小：4.18MB
文件格式：ZIP
更新时间：2024-05-28 00:05:34
JupyterNotebook
使用PyTorch DeepLab v3进行语义分割的转移学习 该存储库包含用于在PyTorch中微调的代码。 该模型来自。 可以在以下位置找到该教程： : 我已经微调了数据集的模型。 该模型经过微调25个纪元，并达到了0.842的测试AUROC值。 该模型在样本图像上的分段输出如下所示。 安装依赖 使用点 pip install -r requirements.txt 使用conda conda env create -f environment.yml 模块的用途 Usage: main.py [OPTIONS] Options: --data-directory TEXT Specify the data directory. [required] --exp_directory TEXT Specify the experiment directory.