Basics
Introduction
Hotkeys shortlist
Tutorials
Getting started with Hasty
How to handle EXIF rotation
How to use Hasty API
Workspaces
Creating a pro workspace
Introduction to workspaces
Projects
Exporting data
Project Overview
Dashboard
AI assistants status overview
Creating and editing a project
- Basics
- Classes and attributes
- File manager
- Taxonomy
- Upload images
- Users and Roles
- Import annotations
- Automated labeling
- Advanced options
- Project Summary
- Webhooks
Mounting buckets
- Mounting buckets tutorial
Project Reports
Annotation environment
Introduction to the annotation environment
When to use which tool
Panning and zooming
Manual and semi-automated tools
- Polygon
- Brush and Eraser
- Polygon <-> Mask Conversion
- Merging polygons/masks into one annotation
- Bounding Box
- DEXTR
- ATOM powered by SAM
- Box to Instance
- Image Tags
- Label Attributes
Move/edit
AI assistants
- Label Class Prediction Assistant
- Label Attribute Prediction Assistant
- Instance Segmentation Assistant
- Object Detection Assistant
- Semantic Segmentation Assistant
- Image Tag Prediction Assistant
Quality assurance and control
AI Consensus Scoring
- AI Consensus Scoring in the Annotation Environment
- AI Consensus Scoring for Image Tags
- AI Consensus Scoring for Attributes
- How to improve your AI Consensus Scoring results
Manual Review
Export formats
COCO Dataset format
Image export
Pascal VOC
Hasty JSON v1.1
Semantic Segmentation (png images)
How to export a project
Import formats
Import annotations (Beta)
- How to import annotations for an Object Detection task
- How to import annotations for an Instance Segmentation task
Supported import types
Model Playground
Accessing Model Playground
Data Split
- Creating a new split
- Split Results
Creating an experiment
- Individual experiment overview
Model Dashboard
Dashboard widgets and visualizations
- Attribute Prediction
- Image tag prediction
- Instance Segmentation
- Label class prediction
- Object Detection
Model Exports
- Deploy
Image Transformations
TorchScript Sample Inference Scripts
- Attribute Prediction
- Classification
- Image-Tagging
- Instance Segmentation
- Object Detection
- Semantic Segmentation
Explainability and Saliency maps
- D-RISE
- Grad-CAM
Active Learning
Active Learning in Hasty
- Margin
- Variance
- Entropy
How to use Active Learning in Hasty
Keypoints
How to work with keypoints in Hasty
Semi-Supervised Learning
Semi-Supervised Learning in Hasty

Semantic Segmentation

Sample inference script for torchscript exported semantic segmentation model

    import torch
import numpy as np
from PIL import Image
import torchvision
import json
import matplotlib.pyplot as plt
import cv2

with open('class_mapping.json') as data:
    mappings = json.load(data)

class_mapping = {item['model_idx']: item['class_name'] for item in mappings}

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

model = torch.jit.load('model.pt').to(device)

image_path = '/path/to/your/image'
image = Image.open(image_path)
# Transform your image if the config.yaml shows
# you used any image transforms for validation data
image = np.array(image)
h, w = image.shape[:2]
# Convert to torch tensor
x = torch.from_numpy(image).to(device)
with torch.no_grad():
    # Convert to channels first, convert to float datatype
    x = x.permute(2, 0, 1).unsqueeze(dim=0).float()
    y = model(x)
    mask = torch.argmax(y, dim=1).squeeze()

# Overlay predicted mask on image and display
plt.imshow(image)
plt.imshow(mask, alpha=0.5)
plt.show()
    python

The script above should produce outputs that look like this:

Example output from the semseg inference script, yellow highlights the present class.

Last modified 6mo ago

Previous - TorchScript Sample Inference Scripts

Object Detection

Next - Model Playground

Explainability and Saliency maps