在程序中的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],
},
}