ECR_REPO=$1
# login to ECR
REGION=$(aws configure get region)
ACCOUNT=$(aws sts get-caller-identity --query Account --output text)
FULLNAME="${ACCOUNT}.dkr.ecr.${REGION}.amazonaws.com/${ECR_REPO}:awsdet"
$(aws ecr get-login --region ${REGION} --no-include-email)

cd ~/SageMaker/deep-learning-models/models/vision/detection/docker
docker build -t ${ECR_REPO}:awsdet -f Dockerfile.sagemaker .

# push image to ECR
docker tag ${ECR_REPO}:awsdet ${FULLNAME}
docker push ${FULLNAME}
echo ${FULLNAME}