Multi-label classification
| image_path | tags |
|---|---|
| images/image_000001.jpg | car man |
| images/image_000002.jpg | happy dog tie |
| images/image_000003.jpg | boat water |
ludwig experiment \
--dataset image_data.csv \
--config_file config.yaml
With config.yaml:
input_features:
-
name: image_path
type: image
encoder: stacked_cnn
output_features:
-
name: tags
type: set