Skip to content

Commit

Permalink
Update data_format and label_list spec
Browse files Browse the repository at this point in the history
  • Loading branch information
longzw1997 authored and BIGBALLON committed Oct 25, 2023
1 parent 84e58e8 commit 06e3368
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,7 @@ config/datasets_mixed_odvg.json # support mixed dataset for both OD and VG
# Training

- Before starting the training, you need to modify the ``config/datasets_vg_example.json`` according to ``data_format.md``.
- The evaluation code defaults to using coco_val2017 for evaluation. If you are evaluating with your own test set, you need to convert the test data to coco format (not the ovdg format in data_format.md), and modify the config to set **use_coco_eval = False** (The COCO dataset has 80 classes used for training but 90 categories in total, so there is a built-in mapping in the code).
- The evaluation code defaults to using coco_val2017 for evaluation. If you are evaluating with your own test set, you need to convert the test data to coco format (not the ovdg format in data_format.md), and modify the config to set **use_coco_eval = False** (The COCO dataset has 80 classes used for training but 90 categories in total, so there is a built-in mapping in the code). Also, update the **label_list** in the config with your own class names like **label_list=['dog', 'cat', 'person']**.


``` bash
Expand Down
4 changes: 2 additions & 2 deletions data_format.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ Visual Grounding datasets employ the 'grounding' field.
"detection": {
"instances": [
{
"bbox": [262,210,323,338],
"bbox": [262,210,323,338], # [x1,y1,x2,y2]
"label": 0,
"category": "dog"
},
Expand All @@ -71,7 +71,7 @@ Visual Grounding datasets employ the 'grounding' field.
"caption": "a wire hanger with a paper cover that reads we heart our customers",
"regions": [
{
"bbox": [20,215,985,665],
"bbox": [20,215,985,665], # [x1,y1,x2,y2]
"phrase": "a paper cover that reads we heart our customers"
},
{
Expand Down

0 comments on commit 06e3368

Please sign in to comment.