return_layers : 需要做fpn的layer的dict; in_channels_list : return_layer對應的channels. 一個簡單的使用例子: import torch import torchvision.
確定! 回上一頁