TEST_IMG = './test_data/test.jpg' DIMENSION = 1000 MODEL_NAME = 'resnet50'