Anyone ran this project in Mac M1 - without Cuda support?

yucornetto / MGMatting

This repository includes the official project of Mask Guided (MG) Matting, presented in our paper: Mask Guided Matting via Progressive Refinement Network

Other

331 stars 46 forks source link

Anyone ran this project in Mac M1 - without Cuda support? #36

Open shiwanlin opened 2 years ago

shiwanlin commented 2 years ago

The codes are hardcoded with Cuda, e.g. in infer.py:

# build model
model = networks.get_generator(encoder=CONFIG.model.arch.encoder, decoder=CONFIG.model.arch.decoder)
model.cuda()