search

bubbliiiing / yolact-pytorch

这是一个yolact-pytorch的库，可用于训练自己的数据集
MIT License
71 stars 12 forks source link

仅加载主干网络权重和加载整个模型权重的训练效果相差很大 #13

Open goodguyjameswin opened 1 year ago

goodguyjameswin commented 1 year ago

作者大大您好！我这边用自己的实例分割数据集做微调训练，发现仅加载主干网络的预训练权重训练（全程解冻），模型的分割精度非常低（相比加载整个模型的预训练权重训练），检测精度（bbox）持平，想知道原因，需要调整什么参数吗？

goodguyjameswin commented 1 year ago

哦还有，我自己的数据集比较小，大概只有五百张图片（增强前）

bubbliiiing commented 1 year ago

yolact也没有那么好拟合，你可能训练的epoch不够

goodguyjameswin commented 1 year ago

yolact也没有那么好拟合，你可能训练的epoch不够

哦哦，我也想过可能是这个原因

goodguyjameswin commented 1 year ago

yolact也没有那么好拟合，你可能训练的epoch不够

通过平移、旋转增强后数据集两千张图片。batch size 设的 8，训练400 epoch

goodguyjameswin commented 1 year ago

yolact也没有那么好拟合，你可能训练的epoch不够

我看代码里有打印建议设置的训练轮数，但是实际运行时没有提示，以为够了，不知道设多少比较合适

bubbliiiing commented 1 year ago

你按那个训练轮数有用吗00