A plugin for Voxel51, integrating it with the annotation platform Segments.ai
plugin_demo.mp4
fiftyone plugins download https://github.com/segments-ai/segments-voxel51-plugin
To use this plugin, you need to provide your Segments.ai API key. This is done by setting it as an environment variable before launching the app:
export SEGMENTS_API_KEY=<YOUR API KEY HERE>
fiftyone app launch
The following table shows the compatible fityone mediatypes with the Segments.ai label types.
Fiftyone media type | Compatible Segments.ai datasets |
---|---|
image | segmentation-bitmap, vector, bbox, keypoints |
pointcloud | pointcloud-cuboid, pointcloud-vector |
3d1 | pointcloud-cuboid, pointcloud-vector |
video | Not supported |
group | Not supported |
You can bring up the operators by pressing "`" and searching for their name. Below you can find the available operators.
This operator selects the corresponding Segments.ai dataset for the current fiftyone dataset. This is needed for the other operators to interact with existing Segments.ai datasets. This operator will show a dropdown list of all your Segments.ai datasets with the corresponding data type. The selected Segments.ai dataset is stored internally in the fiftyone dataset object.
With this operator you can, from the fiftyone app, create a new Segments.ai annotation dataset. You can either upload the whole fiftyone dataset, upload only the current view, or you can upload all of the selected samples.
Current limitations:
- Fiftyone datasets with grouped media, and 3D scenes are not yet supported. If you want to upload 3D pointclouds, please use point cloud datasets
You can fetch annotations from a Segments.ai dataset using this operator. When you call this operator, you can select one of your datasets and one of its releases. It will then pull the annotations and display them within the fiftyone app.
When pulling labels from Segments.ai, the operator will match fiftyone samples with Segments.ai samples using the segments UUID. These are automatically stored in the fiftyone sample object as an attribute, under the key "segments_uuid". If you have uploaded your dataset to Segments.ai using the request_annotations
operator, this is done for you automatically! If not, you will have to provide this information yourself. This can be done as follows:
import fiftyone as fo
dataset = fo.load_dataset("your_dataset")
for sample in dataset.iter_samples(autosave=True,progress=True):
# Somehow match the fo.Sample with the segments.ai sample
sample["segments_uuid"] = "<UUID OF YOUR SAMPLE HERE>"
Alternatively, this example uses FiftyOne's set_values() method to perform a bulk update:
import fiftyone as fo
dataset = fo.load_dataset("your_dataset")
dataset.set_values("segments_uuid",your_values_map,key_field=your_key)
Current limitations:
- It's currently not possible to pull annotations for Segments.ai sequences.
Segments.ai issues are useful mechanisms for communicating problems in the labelling with you annotation team. With this operator, you can file an issue in a specific sample from within the fiftyone app. Simply select 1 sample and run this operator. You will be able to describe your issue in the text box, which will then be uploaded to Segments.ai!
Footnotes
-
request_annotations operator not yet supported for 3d mediatype. Please use the pointcloud mediatype if you need this feature. ↩