Optimize parse_layers.py
Optimize parse_layers.py