Ailia input shape: (1, 3, 512, 736)
All resources in this repository (code, models, demos, etc.) are released under the Creative Commons Attribution NonCommercial ShareAlike 4.0 license. The license will be changed to allow commercial use after our paper is accepted.
Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.
For the sample image,
$ python3 modnet.pyIf you want to specify the input image, put the image path after the --input option.
You can use --savepath option to change the name of the output file to save.
$ python3 modnet.py --input IMAGE_PATH --savepath SAVE_IMAGE_PATHBy adding the --video option, you can input the video.
If you pass 0 as an argument to VIDEO_PATH, you can use the webcam input instead of the video file.
$ python3 modnet.py --video VIDEO_PATHMODNet: Trimap-Free Portrait Matting in Real Time
Pytorch
ONNX opset = 11