Skip to content

Instantly share code, notes, and snippets.

View cosmincatalin's full-sized avatar
👻

Cosmin Catalin SANDA cosmincatalin

👻
View GitHub Profile
@cosmincatalin
cosmincatalin / object-counting-sagemaker-script.py
Created March 21, 2018 10:08
The complete SageMaker script for building an MXNet model that counts shapes in an image.
import base64
import json
import logging
from pickle import load
import mxnet as mx
import numpy as np
from mxnet import autograd, nd, gluon
from mxnet.gluon import Trainer
from mxnet.gluon.loss import L2Loss
@cosmincatalin
cosmincatalin / object-counting-sagemaker-script.py
Created March 20, 2018 14:52
The first part of a SageMaker script for building an MXNet model that counts shapes in an image.
import logging
from pickle import load
import mxnet as mx
import numpy as np
from mxnet import autograd, nd, gluon
from mxnet.gluon import Trainer
from mxnet.gluon.loss import L2Loss
from mxnet.gluon.nn import Conv2D, MaxPool2D, Dropout, Flatten, Dense, Sequential
from mxnet.initializer import Xavier