| { |
| "training": { |
| "model_name": "openai/clip-vit-base-patch32", |
| "num_epochs": 5, |
| "batch_size": 8, |
| "learning_rate": 1e-5, |
| "warmup_steps": 100, |
| "train_split": 0.8, |
| "eval_split": 0.2 |
| }, |
| "data": { |
| "image_dir": "training_data/images", |
| "output_dir": "training_data/trained_models", |
| "supported_formats": ["jpg", "jpeg", "png", "webp"] |
| }, |
| "flower_labels": [ |
| "rose", "tulip", "lily", "peony", "sunflower", "chrysanthemum", "carnation", |
| "orchid", "hydrangea", "daisy", "dahlia", "ranunculus", "anemone", "marigold", |
| "lavender", "magnolia", "gardenia", "camellia", "jasmine", "iris", "gerbera", |
| "zinnia", "hibiscus", "lotus", "poppy", "sweet pea", "freesia", "lisianthus", |
| "calla lily", "cherry blossom", "plumeria", "cosmos" |
| ] |
| } |
|
|
