1.预训练模型
预训练模型是可以直接用的,也可以在此基础上添加自己的数据集进行训练。
1.基于大的数据集训练,可以捕捉通用特征,识别率不会低。适用于通用任务,如果数据集足够大且任务通用,预训练模型可能更适合。
2.在预训练模型基础上进行训练速度也会比自己从零开始训练要快同时也减少了数据需求。
3.可以使用已经训练好的权重,通常对硬件需求较低。
4.预训练模型的迁移学习非常灵活,可以通过微调不同的层来适应不同的任务。
5.适用性广泛,多用于计算机视觉任务,如目标检测、图像分类、语义分割等。
2.自己训练的模型
1.自己训练模型在时间上耗时更多,需要自己收集,处理数据集。
2.自己训练的模型更适合特定、复杂的任务。搭建网络,针对特定任务,可能需要设计适合的神经网络框架。当然,这个也需要更多领域的知识和实验,尤其是在设计网络架构时。
3.自己训练的模型可能更容易过拟合,特别是在小数据集上。
4.自己训练深度学习的模型通常需要更多的计算资源,如GPU、TPU等硬件。