Skip to content

Albumentations KNN augmentation #6

@danellecline

Description

@danellecline

Add albumentations augmentation as an option to the http://doris.shore.mbari.org:8000/docs#/default/knn
endpoint

import albumentations as A
return A.Compose([
    A.RandomResizedCrop(size=(image_size, image_size), scale=(0.5, 1.0), p=1.0),
    A.GaussianBlur(blur_limit=(3, 7), sigma_limit=0.1, p=0.5),
    ToTensorV2()
])

The algorithm should pick the closes by similarity.
This should be an option since it will increase processing time.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions