Bug of localization in train.py

[zhiyuanyou]

Hi,

In train.py, there is a localization option shown like, parser.add_argument('--localization', default='False', choices=('True', 'False'), help='If True train on (64,64) cropped patches') However, if 'False' actually means True because 'False' is seen as string. So, the localization option is always True.

[LilitYolyan]

thanks for noticing, we will fix it asap

[princeofpython]

Also heatmap_on_image() in the final line of code has 3 arguments, but while defining it, there were only 2 arguments.