YOLO | Image Classification
Last updated
Last updated
The YOLO | Image Classification plugin allows you to add classifiers to an image or to individual annotations.
Adding the plugin to your organization will allow you, with one click, to have the plugin add text classifiers to either entire assets or to objects (labels) in an asset.
From the Plugin Directory, search for the name of the plugin and add it to your organization. More information on installing plugins can be found in the Installing Plugins page.
If you are going to use the tool classification mode of this plugin, ensure you have created at least one Bounding Box labeling tool in your project and that you have created the objects you would like the plugin to classify in the image you will pass as input.
If you are going to use the image classification mode of this plugin, ensure you have created at least one text classification class in your project.
Navigate to the project where you'd like to perform the task.
Enter the Settings tab, then the Plugins section.
Find the plugin, and click on Open. A dialog will appear:
Ensure the dot before the plugin's name is green. The dot indicates the plugin's status, where green signifies the plugin's code is running, and red signifies the plugin's code is not running.
You will not be able to run plugins the code of which isn't running.
Do the following:
If you are going to use the image classification mode of the plugin, in the "Class Mapping" field, open the left-hand dropdown, and pick the "Classification Result" option.
If you are going to use the image classification mode of the plugin, open the right-hand dropdown, and pick the text class you'd like the plugin to write the classifications to.
Click on the "plus" button to finalize the pairing.
In the "Asset ID" field, enter the ID of the asset you would like to annotate.
In the "Task ID" field, enter the ID of the task you would like to annotate. (you may find info for steps 4 and 5 by entering the task and opening the "Task Info" drawer on the right-hand side, or by obtaining project information from the SDK.)
Click on Run. The plugin will start detecting the selected object classes in the image.
In the project workflow, add a Plugin-type stage such that it receives the assets you wish to classify as input.
Click on the plugin-type stage and select the plugin, then alter its settings as done in the previous section:
Click on Save to save the workflow. The plugin will now run every time it receives a compatible image asset as input.
With the image labeling editor open, click on the "Plugin Dropdown" to see a list of available model plugins:
If a default configuration was set, click directly on the plugin to run it immediately with the pre-set configuration. Otherwise, click on the three dots next to it to open the Run menu.
From the Run menu, configure its settings and click on Run to run it on the current image.
classify: allows you to pick between image classification mode (asset) and object classification mode (tool)
