pip install flower
import flower
python
from flower.data import DataLoader
data_dir = "/path/to/flower_dataset"
label_file = "/path/to/labels.txt"
image_size = (224, 224)
batch_size = 32
data_loader = DataLoader(data_dir, label_file, image_size, batch_size)
python
from flower.transforms import RandomCrop, Normalize
transform = flower.transforms.Compose([
RandomCrop(224),
Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
processed_data = transform(data_loader.get_data())
python
from flower.features import extract_features
from flower.models import SVM
features = extract_features(processed_data)
labels = data_loader.get_labels()
model = SVM()
model.train(features, labels)
python
from flower.visualization import imshow, plot_histogram
image = processed_data[0]
imshow(image)
feature = features[0]
plot_histogram(feature)