| 21 | |
| 22 | |
| 23 | def get_pretty_layer_name(settings, layer_name): |
| 24 | has_old_settings = hasattr(settings, 'caffevis_layer_pretty_names') |
| 25 | has_new_settings = hasattr(settings, 'caffevis_layer_pretty_name_fn') |
| 26 | if has_old_settings and not has_new_settings: |
| 27 | print ('WARNING: Your settings.py and/or settings_local.py are out of date.' |
| 28 | 'caffevis_layer_pretty_names has been replaced with caffevis_layer_pretty_name_fn.' |
| 29 | 'Update your settings.py and/or settings_local.py (see documentation in' |
| 30 | 'setttings.py) to remove this warning.') |
| 31 | return settings.caffevis_layer_pretty_names.get(layer_name, layer_name) |
| 32 | |
| 33 | ret = layer_name |
| 34 | if hasattr(settings, 'caffevis_layer_pretty_name_fn'): |
| 35 | ret = settings.caffevis_layer_pretty_name_fn(ret) |
| 36 | if ret != layer_name: |
| 37 | print ' Prettified layer name: "%s" -> "%s"' % (layer_name, ret) |
| 38 | return ret |
| 39 | |
| 40 | |
| 41 | def read_label_file(filename): |