|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import sys |
|
import os |
|
|
|
LOCAL_PATH = os.path.dirname(os.path.abspath(__file__)) |
|
TEST_PATH = os.path.join(LOCAL_PATH, "../../../", "test") |
|
sys.path.append(TEST_PATH) |
|
|
|
from paddleseg.utils.download import download_file_and_uncompress |
|
|
|
model_urls = { |
|
"pphumanseg_lite_portrait_398x224_with_softmax": |
|
"https://paddleseg.bj.bcebos.com/dygraph/ppseg/ppseg_lite_portrait_398x224_with_softmax.tar.gz", |
|
"deeplabv3p_resnet50_os8_humanseg_512x512_100k_with_softmax": |
|
"https://paddleseg.bj.bcebos.com/dygraph/humanseg/export/deeplabv3p_resnet50_os8_humanseg_512x512_100k_with_softmax.zip", |
|
"fcn_hrnetw18_small_v1_humanseg_192x192_with_softmax": |
|
"https://paddleseg.bj.bcebos.com/dygraph/humanseg/export/fcn_hrnetw18_small_v1_humanseg_192x192_with_softmax.zip", |
|
"pphumanseg_lite_generic_humanseg_192x192_with_softmax": |
|
"https://paddleseg.bj.bcebos.com/dygraph/humanseg/export/pphumanseg_lite_generic_192x192_with_softmax.zip", |
|
} |
|
|
|
if __name__ == "__main__": |
|
for model_name, url in model_urls.items(): |
|
download_file_and_uncompress( |
|
url=url, |
|
savepath=LOCAL_PATH, |
|
extrapath=LOCAL_PATH, |
|
extraname=model_name) |
|
|
|
print("Export model download success!") |
|
|