trsvchn/coco-viewer

Minimalistic COCO Dataset Viewer in Tkinter

COCO Viewer

Simple COCO Objects Viewer in Tkinter. Allows quick viewing on local machine.

Keyboard Shortcuts

Key	Action
→, J	Next Image
←, K	Previous Image
L, Ctrl + L	Toggle Labels
B, Ctrl + B	Toggle Boxes
M, Ctrl + M	Toggle Masks
Ctrl + S	Save Current Image
Ctrl + Q, Ctrl + W	Exit Viewer

Requirements

python3 PIL

Installation

git clone https://github.com/trsvchn/coco-viewer.git

Usage

python cocoviewer.py -h

usage: cocoviewer.py [-h] [-i PATH] [-a PATH]

View images with bboxes from the COCO dataset

optional arguments:
  -h, --help                    show this help message and exit
  -i PATH, --images PATH        path to images folder
  -a PATH, --annotations PATH   path to annotations json file

Example:

python cocoviewer.py -i coco/images/val/val2017 -a coco/annotations/val/instances_val2017.json

coco-viewer's Issues

Save file is broken

Error when parsing json file

When trying to run the coco-viewer I get error

INFO: Starting...
INFO: Parsing ../SKU110K_fixed/annotations/val.json...
Traceback (most recent call last):
  File "cocoviewer.py", line 745, in <module>
    main()
  File "cocoviewer.py", line 740, in main
    Controller(data, root, image_panel, statusbar, menu, objects_panel, sliders)
  File "cocoviewer.py", line 477, in __init__
    self.update_img()
  File "cocoviewer.py", line 511, in update_img
    label_size=label_size,
  File "cocoviewer.py", line 90, in compose_current_image
    self.compose_image(self.current_image, **kwargs)
  File "cocoviewer.py", line 78, in compose_image
    draw_masks(draw, objects, names_colors, ignore, alpha)
  File "cocoviewer.py", line 194, in draw_masks
    draw.polygon(m[0], outline=fill, fill=fill)
IndexError: list index out of range

My json file contains only bounding boxes and no masks. Looks like the coco-viewer returns error because there are no polygons defined in json file.

Needed to install DejaVuSans.ttf

To get this script to work, I needed to add DejaVuSans.ttf to the coco-viewer directory. I dropped it in both the root and the assets directory. The script did not find it after I installed on Windows, and my conda environment did not have it by default.

I downloaded it from here: https://www.fontsquirrel.com/fonts/dejavu-sans

Thank you for the script!

Large image support

Hello, is large image support included? Are we supposed to be able to zoom out images?

Can't have a background annotation

It should be possible to have the first element of a segmentation RLE element to be 0 to be able to label the background but it doesn't seem to be supported. I've tried to insert an arbitrary value in the counts element and reducing the corresponding amount in the next element but it doesn't work either.

It works correctly by appending 0 to the counts element in FiftyOne :

KeyError: 'iscrowd'

Hi,
I am getting this error when I run coco-viewer.

(base) chris@Chriss-MacBook-Pro coco-viewer % python cocoviewer.py -i /Users/chris/Documents/video_frames/2022-07-22/Over -a /Users/chris/Downloads/test.json INFO: Starting... INFO: Parsing /Users/chris/Downloads/test.json... Traceback (most recent call last): File "/Users/chris/Documents/coco-viewer/cocoviewer.py", line 821, in <module> main() File "/Users/chris/Documents/coco-viewer/cocoviewer.py", line 816, in main Controller(data, root, image_panel, statusbar, menu, objects_panel, sliders) File "/Users/chris/Documents/coco-viewer/cocoviewer.py", line 514, in __init__ self.update_img() File "/Users/chris/Documents/coco-viewer/cocoviewer.py", line 573, in update_img self.compose_image( File "/Users/chris/Documents/coco-viewer/cocoviewer.py", line 542, in compose_image draw_masks(draw, objects, names_colors, ignore, alpha) File "/Users/chris/Documents/coco-viewer/cocoviewer.py", line 200, in draw_masks elif isinstance(m, dict) and objects[i]["iscrowd"]: KeyError: 'iscrowd' (base) chris@Chriss-MacBook-Pro coco-viewer %

I am just trying out this tool to view my masks. This specific example I'm looking at one image. Since I can't attach json files, I attached a txt file, that I copied and pasted from the json file.
test_json.txt

I'm running this on MacOS Monterey on the M1 chip.

Any help would be greatly appreciated.
Thanks

[research] How to test Tkinter app

It would be great to add test suite and CI. But I don't really know how to test python gui app.

Add view menu

Currently, app draws a bboxes with masks.

Feature request:

Add a view menu to control the appearance of masks and bounding boxes.

JSON file question

hi @trsvchn, does segmentation element in JSON file needs to be written in RLE format, polygon or it can be just pixel coordinates of my ground truth?

Refactoring the app

I think it is better to have single-file app, since code base is not so HUGE!

Implement statusbar

The idea is to add a simple status bar somewhere on the bottom, to show image name, number of objects, maybe classes, etc

Scrolling for ImageWidget

Add a scrolling for ImageWidget. This can be achieved with the Canvas widget, since Frame or Label is not scrollable.

Sliders are always active

Deactivate sliders when toggling or using View menu

Issue: PIL deprecated the textsize() method in 10.0.0

cocoviewer.py, line 183:
tw, th = draw.textsize(text, font)
PIL deprecated the textsize() method in 10.0.0:
https://pillow.readthedocs.io/en/stable/releasenotes/10.0.0.html

Proposed fix:

x1, y1, x2, y2 = font.getbbox(text)
tw, th = (x2-x1, y2-y1)

Add a option to control bbox thickness and mask opacity

Handle case with empty args during launching

Currently app requires path to image folder and annotations file to launch.

Add support to launch the app with warning dialog about empty data.

Show a list of present objects and categories for the current image

Add a box/label with a listed objects and classes,
Add option to interact with them.

Add "select a file/folder" support from the app

Now the only way to select the data/annotations the app is cli.

Let's add a option to select them from File menu.

Add menubar

Add a simple menu
Related to #1

Set View menu togglers to act globally

Current view menu affects on local image, but it should be a global feature.

Usage idea:

• use View menu to set up things globally
• use keyboard to toggle bboxes/masks on current image

Handle RLE masks

Now only polygon masks are supported. We need to handle RLE masks as well.

Model is too much dataset oriented

Improve Data (split) class to handle single image cases.

Add Class Labels

We need this.

How to change the loaded image?

I have 100 images. This software can't let me change the image, though it shows 1/100 at the bottom right corner.

Edit: It is possible from keyboard. Left and Right arrow. Do we need better instruction for users like me?

Font labels and other platforms

How to set a font correctly to fit every platform?

Feature request: instance (object) coloring

In addition to category based coloring implement object based coloring: each object with separate color

How to?

• Extract color preparing logic
• Add option to Menu

cc @psimadiris

Split ImageWidget into model and controller

Refactor Main ImageWidget to fit MVC.

App layout is not statusbar friendly

Current layout is great, but status bar is clipped when resizing.

no display name and no $DISPLAY environment variable

Showing the following error:
INFO: Starting... Traceback (most recent call last): File "./coco-viewer/cocoviewer.py", line 898, in <module> main() File "./coco-viewer/cocoviewer.py", line 877, in main root = tk.Tk() File "/usr/lib/python3.7/tkinter/__init__.py", line 2023, in __init__ self.tk = _tkinter.create(screenName, baseName, className, interactive, wantobjects, useTk, sync, use) _tkinter.TclError: no display name and no $DISPLAY environment variable

Migrate to ttk

Let's replace current widgets with ttk widgets. It will help with styling in a future.

Feature request: save file

This is a nice and simple viewer for the coco annotations. It would be nice if there was an option to save the file when looking at them.

cannot open resource

Sliders and Toggles interaction is wierd

This is related to local and global state and update_img called differently.

Bug to fix in default 'info'

cocoviewer.py
Original, line 542

self.description_status.set(f"{self.data.instances.get('info', '').get('description', '')}")

Should be

self.description_status.set(f"{self.data.instances.get('info', dict()).get('description', '')}")

Transfer image composition from PIL to Canvas

From #45 we are using Canvas to display an image. Let's use the power of it!

n_objects=80 is hardcoded in method get_categories.

The function get_categories() can be modified to get n_objects = len(instances['categories'])

So that if someone is using more than 80 classes he can use the tool, without any issues.