u2net-saliency / config.json
reeteshmukul's picture
Update config.json
82f8955
raw
history blame
170 Bytes
{
"model_type" : "u2net",
"image_processor_type" : "u2net",
"architectures": [
"U2NetForSaliencyEstimation"
],
"input_channels": 3,
"output_channels": 1
}