Python script for take image size