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