mirror of
https://git.mirrors.martin98.com/https://github.com/xuebinqin/U-2-Net
synced 2025-04-23 22:29:58 +08:00
14 lines
431 B
Python
14 lines
431 B
Python
import os
|
|
import gdown
|
|
|
|
os.makedirs('./saved_models/u2net', exist_ok=True)
|
|
os.makedirs('./saved_models/u2net_portrait', exist_ok=True)
|
|
|
|
gdown.download('https://drive.google.com/uc?id=1ao1ovG1Qtx4b7EoskHXmi2E9rp5CHLcZ',
|
|
'./saved_models/u2net/u2net.pth',
|
|
quiet=False)
|
|
|
|
gdown.download('https://drive.google.com/uc?id=1IG3HdpcRiDoWNookbncQjeaPN28t90yW',
|
|
'./saved_models/u2net_portrait/u2net_portrait.pth',
|
|
quiet=False)
|