我已经能够重新训练对象检测模型以适用于我的自定义类别。但现在我想在推理过程中将输入修改为不同的颜色空间(最好是YUV)。
我需要修改网络吗?可以使用相同的API来完成吗?我仍然可以使用迁移学习,还是必须从头开始训练?
回答:
最简单的做法可能是直接转换为RGB,但使用迁移学习进行训练肯定是可行的。你仍然需要将YUV值保存为JPEG的RGB通道,这样做的优势是否显著还不清楚,与直接转换为RGB相比。
一个可能的小速度优势是,如果你想以较低的空间分辨率表示色度(U,V)通道(这通常是常见的做法),但这可能需要设计一个能够利用这一优势的自定义架构。