search

codecat0 / CV

本仓库将使用Pytorch框架实现经典的图像分类网络、目标检测网络、图像分割网络，图像生成网络等，并会持续更新！！！
214 stars 36 forks source link

backbone训练 #5

Open Pillowmm opened 1 year ago

Pillowmm commented 1 year ago

博主 请问如果要选择其中之一作为backbone进行训练得到预训练权重文件，是不是只要把输出通道数和分类的代码改掉就可以了？