You can deploy this model and start earning money today!

Yodazon/
3DPrintFailureType

Name: 3DPrintFailureType
Rating: 2.5 (8 reviews)

Views

No views yet

This model is to help determine the type of problem a 3D print has. The model uses AlexNet CNN Architecture built using PyTorch

The model trained on images of 3D prints as they are printing as well as post printing. Training set of images is about ~5GB

Current version has 4 outputs:

Good
Spaghetti
Stringing
Overextrusion

Of its current iteration, the Model can not determine during an inference if the input is an actual 3D Print or Not.

Future updates will include

Determine if the image is a 3D print or not
Determine if the image is during printing or once complete

To make an inference

Classes

class_names = {0: 'good', 1: 'spaghetti', 2: 'stringing', 3: 'underextrusion'}

Pre-Process the image using the following python function

def preProcess(image):
    # Open the image from raw bytes
    image = Image.open(BytesIO(image)).convert('RGB')

    transform = transforms.Compose([
        transforms.Resize(227),
        transforms.CenterCrop(227),
        transforms.ToTensor(),
        transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
    ])

    input_image = transform(image).unsqueeze(0)
    return input_image

Yodazon/3DPrintFailureType

Yodazon/
3DPrintFailureType