Skip to content

Commit c201e38

Browse files
authored
addind_queue_dataset
Based on the prediction import the expected project queue set is currently should be dataset (since the model is querying dataset to make prediction on). This would need to be rework for batch(es)
1 parent 0ca5240 commit c201e38

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/integrations/detectron2/coco_object.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,7 @@
111111
"import labelbox as lb\n",
112112
"import labelbox.types as lb_types\n",
113113
"from labelbox.data.serialization import COCOConverter\n",
114+
"from labelbox.schema.queue_mode import QueueMode\n",
114115
"\n",
115116
"from labelbox.data.metrics import (\n",
116117
" feature_miou_metric, \n",
@@ -544,7 +545,7 @@
544545
"metadata": {},
545546
"source": [
546547
"# Create a new project for upload\n",
547-
"project = client.create_project(name=\"detectron_mal_project\", media_type=lb.MediaType.Image)\n",
548+
"project = client.create_project(name=\"detectron_mal_project\", media_type=lb.MediaType.Image, queue_mode=QueueMode.Dataset)\n",
548549
"editor = next(\n",
549550
" client.get_labeling_frontends(where=lb.LabelingFrontend.name == 'editor'))\n",
550551
"project.setup(editor, labels_mal.get_ontology().asdict())\n",

0 commit comments

Comments
 (0)