Commit Graph

26 Commits (8d6bf1d332df527a389b620eb42e9375e98e453d)

Author SHA1 Message Date
Linyiqi a8b528de3c
[Enhance] Formatting non distributed training and inference and Supporting CPU training. (#42)
* [Docs] update batch size

* Fix bug in non-distributed multi-gpu training/testing

* support cpu training

* update cpu training and testing
2022-03-11 00:25:34 +08:00
Linyiqi 66d92a5abb
[Enhancement] Set Num thread to 0 (#36)
* [Docs] update batch size

* [Enhancement] Set Num thread to 0

* set omp mkl threads numbers

* set cls omp mkl threads numbers

* add comments
2022-01-27 20:19:14 +08:00
Linyiqi 217aff67f7
[Docs] Update introduction and fix issues (#10)
* fix link issue

* fix link issue

* [Docs] update intro and fix issues

* [Docs] fix translation

* [Docs] update introduction

* remove duplicated model_zoo.yml

* update intro and fix codecov icon

* update README
2021-11-26 16:15:33 +08:00
Linyiqi 486c8c2fd7
[Docs] fix typos and invalid link (#7)
* fix link issue

* fix link issue
2021-11-24 16:20:29 +08:00
Linyiqi 92199add7e
Update docs and README (#56)
* fix init

* fix test api

fix test api bug

* add metarcnn fsdetview config

* udpate faq.md

* fix md

* udpate docs config

* update data preparation

* udpate demo README
2021-11-22 16:43:26 +08:00
Linyiqi acd0ab8251
update config & README & miscellaneous typos (#54)
* update all the code

* update config

* update image and README

* update README

* update README and metafile

* update RAEDME
2021-11-22 00:22:07 +08:00
Linyiqi 0193ce1443
Update resource & add mim & update tools script (#51)
* update resource & env & fix logger issue

* update mim
2021-11-12 23:27:20 +08:00
Linyiqi 3508afca93
Reformat config files (#46)
* fix init

* fix test api

fix test api bug

* add metarcnn fsdetview config

* update pr

* update meta file

* fix CI
2021-11-01 23:27:01 +08:00
Linyiqi 37cd8a1f7d
Fix docs and add test code (#45)
* fix init

* fix test api

fix test api bug

* add metarcnn fsdetview config

* add pr

* add metatestparallel comments

* add test code and fix typos

* add test code of model frozen

* update test det forward code

* update pr

* update doc str
2021-11-01 16:33:21 +08:00
Linyiqi 8776d7edf2
fix code style (#44)
* add pr

* add metatestparallel comments
2021-10-23 10:14:08 +08:00
Linyiqi ae96fa303c
Update lib version in requeriments and fix some bugs (#40) 2021-10-11 19:02:26 +08:00
Linyiqi b3b2cd598e
unified interface of train.py (#38)
* update train file

* fix comments
2021-10-11 18:14:26 +08:00
Linyiqi 4e28607ced
add multiple run and visualize scripts (#33)
* fix init

* fix test api

fix test api bug

* add metarcnn fsdetview config

* add multi run

* update docs

* rename file
2021-09-26 15:45:54 +08:00
Linyiqi fdb8e999ed
upgrade version (#34)
* upgrade version

* fix comments
2021-09-26 14:24:52 +08:00
Linyiqi 93d5899ace
fix test api bug (#29) 2021-09-03 21:10:23 +08:00
Linyiqi a55d76ff8a
add fewshot classification v2 (#27)
* bug fix

* fix comments

* fix doc str

* seprate the cls and det

* update script

* fix commnets

* fix commnets

* fix commnets

test 0 loss

fix commnets

* update tools

* rename detectors

* rename detectors

* rename detectors

* rename detectors

* del frcnn config

* update doc str

* update config

* fix comments

* fix comments

* update config

* update config

* update config

* update config

* fix comments

* fix comments

* fix comments

* fix comments

* add mpsr

* update config

* update config

* test bug voc data

* mv vis to misc

* classification framwork

* add fewshot classification methods

* set before method static

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* add blank line for bar

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* Revert "update config"

This reverts commit a71c6aae288f282887c8d75875b82333ada20658.

* Revert "update config"

This reverts commit d21d5bcccf00195968ccd7ea1952672a54e13ac1.

* Revert "update config"

This reverts commit 1786c07902ec885f0f456cca39cd47872c5eaf64.

* Revert "update config"

This reverts commit 27efc60b04aa96c45c5ee02307014d51518b6805.

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* add resnet12

* update config

* update config

* update config

* update config

* update config

* update config and backbone

* update config and backbone

* update config and backbone

* update config and backbone

* update config and backbone

* update config and backbone

* fix dataset sampling

* fix dataset sampling

* add baseline config

* update  baseline config

* update  baseline++ config

* update  baseline++ config

* update  baseline++ config

* update  baseline config

* fix pipeline

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update test backbone

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* fix maml bug

* fix maml bug

* update config

* fix maml bug

* update config

* update config

* update config

* fix maml bug

* update config

* update config

* update config

* update config

* update config

* update config

* add tiered imagenet

* update config

* update config

* update config

* update config

* add dataloader

* add dataloader

* add dataloader

* add dataloader

* add dataloader

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* add doc

* add doc

* fix tiered imagenet

* fix tiered imagenet

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* fix comments

* fix comments

* update config

* update config

* update config

* update config

* update config

* update config

* fix eval prior

* update relation net

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update doc str

* update doc str

* update doc str

* update config

* update config

* update config

* update config

* update doc str

* add few shot classification

* update meta baseline config

* add few shot classification

* update script

* update script

* update script

* update script

* update script

* update config

* update config

* update config

* update config

* update margin to 0

* update config

* fix comments

* fix comments

* update config

* update config

* update config

* fix comments

* update config

* fix meta baseline

* fix meta baseline

* update config

* fix meta baseline

* update config

* update config

* fix comments

* fix comments

* update config

* add config

Co-authored-by: zhangshilong <2392587229zsl@gmail.com>
2021-09-01 19:46:51 +08:00
Linyiqi c87e2d8839
fix code style and grammar issues. (#23)
* fix comments
2021-08-02 20:45:49 +08:00
Linyiqi a7c30ad01c
seprate the cls and det (#20)
* create pr

* sperate cls and det

* create pr

* update tools

Co-authored-by: zhangshilong <2392587229zsl@gmail.com>
Co-authored-by: Shilong Zhang <61961338+jshilong@users.noreply.github.com>
2021-08-01 12:26:46 +08:00
Linyiqi 21942a717c
implementation of FSCE (#18)
* fix dataloader bug

* update config

* check rank

* check rank

* check rank

* check rank

* check rank

* check rank

* check rank

* check rank

* rm check rank

* dataset refactoring

* add save dataset function

* add doc string

* update config

* update config

* fix dataset bulider bug

* update ckpt_surgery script

* update ckpt_surgery script

* rename arpn to attention_rpn

* update tfa config

* update tfa config

* update tfa config

* update tfa config

* update tfa config

* update tfa config

* update script

* update config

* update fsdetview config

* update fsdetview config

* update fsdetview config

* update tfa config

* update tfa config

* update fsdetview config

* run script

* run script

* run script

* fix comments

* fix comments

* fix comments

* fix save dataset bug

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update dataset doc

* fix dataset loading bug

* fix dataset loading bug

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* fix eval class splits bug

* update attention rpn voc config

* update attention rpn voc config

* update attention rpn voc config

* update attention rpn config

* fix voc dataset bug

* udpate config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* add dataset visualization

* rm unused arg

* update config

* add visual dataset

* fix dataset loading bug

* update config

* update config

* add visualize dataset code

* voc base training debug

* update tfa voc base config

* update config

* update config

* add multiple training

* update tfa voc lr

* update config

* update config

* update config

* update config

* update config

* update voc metric

* fix voc metric

* add dataset generate code

* update base training

* update save dataset

* update doc string

* update custom hook

* fix dataset loading bug

* test multi-train

update config

update config

update config

update multi train

test multiple train

test multi train

update config

update config

update config

test multi train

test multi train

test multi train

test multi train

test multi train

fix random bug

test multi train

* rm repeat args

* fix bash

* fix seed

* update config

* set random seed

* update config

* fix random seed bug

* update config

* update config

* update config

* update config

* update sh script

* update script

* update multi run

* save support set for queryawaredtaset

* update config

* multi test

* update multi test script

* add meta rcnn model & config

* add config

* update visualizer

* update config

* update config

* test seed

* Revert "test seed"

This reverts commit 803b4af70f000aee1bad32d4da91dc4f2078f138.

* fix seed

* test seed

* update config

* update config

* rm seed test

* update config

* update config

* fix comments

* fix comments

* update config

* fix comments

* update config

* fix comments

fix pipeline parameter

* fix comments

* refactoring ann_file

* fix copy

* fix commets

* fix multi train test

* add config check

* refactoring ann_cfg datasetwrapper

* add doc string

* fix bug

* check copy op

* add dataset name & fix doc str

* fix doc str

* fix doc str

* fix doc str

* fix doc str

* fix doc str

* update tfa

* update fsce config

* update attention_rpn config

* update fsdetview config

* update attention rpn doc

* update metarcnn roihead

* refactoring meta rcnn and fsdetview

* update tfa

* update config

* refactoring models

* refactoring models

* fix comments

* fix module

* update doc str

* update script

* update script

* bug fix

* update doc str

* update doc str

* update config

* update losses

* fix comments

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* fix doc str

* fix doc str

* fix doc str

* update config

* update config

* test losses

* update fsdetview

* update fsdetview

* update config

* update config

* fix typo

* fix doc

* fix comments

* fix comments

* fix doc str

* fix doc str

* update fsdetview

* fix doc str

* update attention rpn

* fix doc

* fix get_classes (#10)

* fix get_classes

* change to split id

* fix pipeline and doc str

* update config

* fix doc str

* update script

* update script

* fix comments

* fix doc str and typo

* fix typo

* update config

* update config

* update config

* update config

* fix bug in fsdetview

* update doc str

* update config

* fix comments

* fix comments

* update fsce

* update fsce

* update fsce

* update multi test

update test

* update attention rpn

* bug fix

* create pr

* fix comments

* fix doc str

* fix commnets

* fix commnets

* fix commnets

test 0 loss

fix commnets

Co-authored-by: zhangshilong <2392587229zsl@gmail.com>
Co-authored-by: Shilong Zhang <61961338+jshilong@users.noreply.github.com>
2021-07-31 23:57:35 +08:00
Linyiqi a643f7ee9b
implementation of TFA (#13)
* update script

* update script

* fix comments

* create pr

* fix comments

Co-authored-by: zhangshilong <2392587229zsl@gmail.com>
Co-authored-by: Shilong Zhang <61961338+jshilong@users.noreply.github.com>
2021-07-30 16:21:43 +08:00
Linyiqi 7ac7436d11
dataset refactoring v2 (#10)
* add doc string

* update configs

* update caffe config

* update caffe config

* update config

* rm ignore ann in fewshot

* add difficult option in voc dataset

* add difficult option in voc dataset

* add difficult option in voc dataset

* show datasets

* add FSCE file

* fix FSCE bug

* update config

* update config

* update config

* update config

* update config

* update config

* run script

* update config

* update config

* test aug

* test aug

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* fix comment

* fix comment

* fix comment

* change voc index

* update config

* change voc iou

* change voc iou

* change voc iou

* change voc iou

* Revert "change voc iou"

This reverts commit e5f4ca8bc4701c7c69e900b746eaed14cccf87bb.

* reset voc

* Revert "Merge branch 'lyq-fsce' of https://github.com/linyq17/mmfewshot into lyq-fsce"

This reverts commit 1fde859a1625eb0cb900e4a9baa1125c2f387ae5, reversing
changes made to 75c40006a23f456a56b688a63e66f133d064e01d.

* update weight_decay

* add fsod training code

* fsod training code

fsod training code

fsod training code

fsod training code

fsod training code

fsod training code

* disable group config

* fix empty gt bug

fix empty gt bug

fix empty gt bug

* fsod test code

* add support template init into test code

* add support template init into test code

* fix coco del bug

* fix coco del bug

* update training config

* update test config

* test config

* test config

* test config

* update anchor config

* update anchor config

* update get bbox

* update get bbox

* update training config

* update loss weight

* update loss weight

* update loss weight

* update training config

* update training config

* add repeat dataset into dataset warpper

* mv base code to fsod

* update config

* update config

* update config

* update docstr of script

* update config of cl branch

* fix few shot config bug

* add docstr

* disable filp

* update weight decay config

* add docstr and update loss weight

* add docstr and update loss weight

* add docstr and update loss weight

* update loss weight

* update loss weight

* update loss weight

* fix arpn bug

* update lr config

* fix test bug

* update config

* update config

* update config

* update config

* update config

* fix support order bug

* fsdetview training and testing code

* update config name

* update config

* update config

* update config

* check data loader

* update config

* update config

* update config

* fix dataloader bug

* update config

* check rank

* check rank

* check rank

* check rank

* check rank

* check rank

* check rank

* check rank

* rm check rank

* dataset refactoring

* add save dataset function

* add doc string

* update config

* update config

* fix dataset bulider bug

* update ckpt_surgery script

* update ckpt_surgery script

* rename arpn to attention_rpn

* update tfa config

* update tfa config

* update tfa config

* update tfa config

* update tfa config

* update tfa config

* update script

* update config

* update fsdetview config

* update fsdetview config

* update fsdetview config

* update tfa config

* update tfa config

* update fsdetview config

* run script

* run script

* run script

* fix comments

* fix comments

* fix comments

* fix save dataset bug

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update dataset doc

* fix dataset loading bug

* fix dataset loading bug

* create dataset pr

* create dataset pr

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* fix eval class splits bug

* fix eval class splits bug

* update attention rpn voc config

* update attention rpn voc config

* update attention rpn voc config

* update attention rpn config

* fix voc dataset bug

* fix voc dataset bug

* udpate config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* update config

* add dataset visualization

* rm unused arg

* rm unused arg

* update config

* add visual dataset

* fix dataset loading bug

* fix dataset loading bug

* update config

* update config

* add visualize dataset code

* voc base training debug

* update tfa voc base config

* update config

* update config

* add multiple training

* update tfa voc lr

* update config

* update config

* update config

* update config

* update config

* update voc metric

* fix voc metric

* add dataset generate code

* update base training

* update save dataset

* update doc string

* create pr

* create pr

* fix dataset loading bug

* fix comments

* save support set for queryawaredtaset

* fix comments

* fix comments

fix pipeline parameter

* fix comments

* refactoring ann_file

* fix commets

* add config check

* refactoring ann_cfg datasetwrapper

* add doc string

* fix bug

* fix bug

* add dataset name & fix doc str

* fix doc str

* fix doc str

* fix doc str

* fix doc str

* rm model config
2021-07-23 12:58:12 +08:00
zhangshilong 05701f9cfe remove formart results 2021-05-16 20:15:07 +08:00
zhangshilong fec56dceef fix test for cls 2021-05-16 20:06:46 +08:00
zhangshilong 314a2d1555 add requirments and remove fuse bn in test 2021-05-16 19:15:21 +08:00
zhangshilong 6726507c6f add suppoort for cls 2021-05-16 16:29:24 +08:00
zhangshilong d1894d07f0 add suppoort for det 2021-05-15 22:47:59 +08:00