debugging
Browse files- livermask/livermask.py +2 -1
livermask/livermask.py
CHANGED
@@ -40,7 +40,7 @@ def post_process(pred):
|
|
40 |
|
41 |
def get_model():
|
42 |
url = "https://drive.google.com/uc?id=12or5Q79at2BtLgQ7IaglNGPFGRlEgEHc"
|
43 |
-
output = "model.h5"
|
44 |
md5 = "ef5a6dfb794b39bea03f5496a9a49d4d"
|
45 |
gdown.cached_download(url, output, md5=md5) #, postprocess=gdown.extractall)
|
46 |
|
@@ -48,6 +48,7 @@ def func(path, output, cpu):
|
|
48 |
|
49 |
cwd = "/".join(os.path.realpath(__file__).replace("\\", "/").split("/")[:-1]) + "/"
|
50 |
|
|
|
51 |
name = cwd + "model.h5"
|
52 |
|
53 |
# get model
|
|
|
40 |
|
41 |
def get_model():
|
42 |
url = "https://drive.google.com/uc?id=12or5Q79at2BtLgQ7IaglNGPFGRlEgEHc"
|
43 |
+
output = "./model.h5"
|
44 |
md5 = "ef5a6dfb794b39bea03f5496a9a49d4d"
|
45 |
gdown.cached_download(url, output, md5=md5) #, postprocess=gdown.extractall)
|
46 |
|
|
|
48 |
|
49 |
cwd = "/".join(os.path.realpath(__file__).replace("\\", "/").split("/")[:-1]) + "/"
|
50 |
|
51 |
+
print("Current cwd:", cwd)
|
52 |
name = cwd + "model.h5"
|
53 |
|
54 |
# get model
|