GitXplorerGitXplorer
l

Thunder-Detr

public
12 stars
1 forks
1 issues

Commits

List of commits on branch master.
Verified
5e5052fda107c671c118e941afd621f90b242300

Update README.md

llessw2020 committed 4 years ago
Verified
949aba6c736015840b2d031e160921068146f343

Update README.md

llessw2020 committed 4 years ago
Verified
b55a041c1980fdec6d9147e777705d3c1e1c2a25

Update README.md

llessw2020 committed 4 years ago
Verified
5bc2ffdeac833322b92f3b83fad16a94b90cfbd1

Add files via upload

llessw2020 committed 4 years ago
Verified
1d1faf12b8c0bd79ae25754e253688d1c8167ad5

Create README.md

llessw2020 committed 4 years ago
Verified
57ae0858e97a8bf1c01c5011a46874001564b0d4

two utils for working with custom coco json files

llessw2020 committed 4 years ago

README

The README file for this repository.

Thunder-Detr

(unofficial) - customized fork of DETR, optimized and tuned for intelligent obj detection on 'real world' custom datasets

This is a customized framework based on FB AI's DETR, but with a number of improvements/modifications to optimize it for obj detection on your own datasets. I had started a colab to show how to do this, but with more and more customizations piling up requiring more and more code modifications to DETR core...it became clear building a codebase focused on handling custom datasets would be better and faster for all. Thus, Thunder-Detr was born, 8/1/2020.

Updates: 8/22/20 - add thunder_file_utils.py. Adds a coco_compressor to remap category_ids to contiguous values and rebase to zero. Adds a show_catids to view the categories in a json file and shows the proper "num_classes" value for training DETR with. Usage:

Various changes to improve results built into Thunder-Detr: 1 - recommend LaProp optimizer vs AdamW. 2 - recommend bs of 4 (vs default 2 in DETR) 3 - recommend ciou over giou default of DETR 4 - recommend additional augmentations esp colorjitter