SSD输入单通道图像的解决方法

在程序中的train_transform_param和test_transform_param中插入‘force_color’:True。具体代码如下所示：

train_transform_param={
        'mirror': True,
        'mean_value':[104,117,123]
        'force_color': True,#添加
        'resize_param':{
                'prob'：1,
                'resize_mode':P.Resize.WARP,
                'height': resize_height,
                'width': resize_width,       
                 'interp_mode':[
                         P.Resize.LINEAR,
                         P.Resize.AREA,
                         P.Resize.NEAREST,
                         P.Resize.CUBIC,
                         P.Resize.LANCZOS4,
                         ],
                 },
          }




test_transform_param={
        'mean_value':[104,117,123]
        'force_color': True,#添加
        'resize_param':{
                'prob'：1,
                'resize_mode':P.Resize.WARP,
                'height': resize_height,
                'width': resize_width,       
                 'interp_mode':[P.Resize.LINEAR],
                 },
          }