Skip to content

Commit

Permalink
Merge pull request yu4u#92 from yu4u/hotfix/weight_url
Browse files Browse the repository at this point in the history
Hotfix/weight url
  • Loading branch information
yu4u authored Apr 15, 2019
2 parents 9410a57 + 11aefce commit ba6e38c
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 6 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -214,8 +214,8 @@ Please refer to [here](appa-real) for the details of the APPA-REAL dataset.
The results of pretrained model is:
```
MAE Apparent: 6.06
MAE Real: 7.38
MAE Apparent: 6.47
MAE Real: 7.61
```
The best result reported in [5] is:
Expand Down
8 changes: 4 additions & 4 deletions evaluate_appa_real.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,16 +8,16 @@
from wide_resnet import WideResNet
from keras.utils.data_utils import get_file

pretrained_model = "https://github.com/yu4u/age-gender-estimation/releases/download/v0.5/weights.18-4.06.hdf5"
modhash = '89f56a39a78454e96379348bddd78c0d'
pretrained_model = "https://github.com/yu4u/age-gender-estimation/releases/download/v0.5/weights.28-3.73.hdf5"
modhash = "fbe63257a054c1c5466cfd7bf14646d6"


def get_args():
parser = argparse.ArgumentParser(description="This script evaluate age estimation model "
"using the APPA-REAL validation data.",
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument("--weight_file", type=str, default=None,
help="path to weight file (e.g. weights.18-4.06.hdf5)")
help="path to weight file (e.g. weights.28-3.73.hdf5)")
parser.add_argument("--depth", type=int, default=16,
help="depth of network")
parser.add_argument("--width", type=int, default=8,
Expand All @@ -33,7 +33,7 @@ def main():
weight_file = args.weight_file

if not weight_file:
weight_file = get_file("weights.18-4.06.hdf5", pretrained_model, cache_subdir="pretrained_models",
weight_file = get_file("weights.28-3.73.hdf5", pretrained_model, cache_subdir="pretrained_models",
file_hash=modhash, cache_dir=os.path.dirname(os.path.abspath(__file__)))

# load model and weights
Expand Down

0 comments on commit ba6e38c

Please sign in to comment.