FireClassification icon indicating copy to clipboard operation
FireClassification copied to clipboard
verified publisher icon fire717

灰度图像的处理

Open springsZ opened this issue 2 years ago • 1 comments

大佬我想请教一下 灰度图像也可以用这个框架直接算吗，需要做修改吗

springsZ avatar Oct 16 '23 02:10 springsZ

当然可以，但是需要手动修改。在加载好预训练模型之后，手动把第一层输入conv的通道数改成1（原始是3）

fire717 avatar Oct 18 '23 08:10 fire717