mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
967 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

15 Commits (a4c219e05d3ab78c20b9d22dedde7dded6fd206c)

Author SHA1 Message Date
Yixiao Fang 1ee9bbe050
[Docs] Update links (#1457) 
* update links

* update readtherdocs

* update

* update

* fix lint

* update

* update

* update

* update cov branch

* update

* update

* update
 2023-04-06 20:58:52 +08:00
Ma Zerun 4f5b38f225
[Refactor] Update almost tools and add unit tests for these tools. (#1418) 
* [Refactor] Update almost tools and add unit tests for these tools.

* Fix Windows UT.
 2023-03-17 10:50:51 +08:00
mzr1996 0979e78573 Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
Ma Zerun a3f2effb17
[Feature] Add `ImageClassificationInferencer`. (#1261) 
* [Feature] Add ImageClassificationInferencer.

* Update inferencer implementation and add unit tests.

* Update documentations.

* Update pre-commit hook

* Update docs
 2023-02-08 14:30:12 +08:00
Ma Zerun 97c4ae8805
[Improve] Update registries of mmcls. (#1306) 
* [Improve] Update registries of mmcls.

* Update according to comments
 2023-01-11 15:20:51 +08:00
Ma Zerun 12d8c78222
[Docs] Refactor get started docs and fix the demo and docker file. (#990) 
* [Docs] Refactor get started docs and fix the demo and docker file.

* Improve docs according to comments
 2022-08-26 18:03:18 +08:00
Ma Zerun 17b24a8230
[Improve] Update apis and packages. (#921) 
* Remove mmcls/core folder

* update dependency version.

* Update `apis/inference.py` and demo.
 2022-07-22 10:59:08 +08:00
Ma Zerun 3e198e63e8
[Docs] Update install tutorials. (#854) 
* [Docs] Update install tutorials.

* [Docs] Improve dataset docs

* Add option to show the results in demo.

* fix typo
 2022-06-01 18:31:57 +08:00
Hu Di b4eefe4794
[Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723) 
* implement training and evaluation on IPU

* fp16 SOTA

* Tput reaches 5600

* 123

* add poptorch dataloder

* change ipu_replicas to ipu-replicas

* add noqa to config long line(website)

* remove ipu dataloder test code

* del one blank line in test_builder

* refine the dataloder initialization

* fix a typo

* refine args for dataloder

* remove an annoted line

* process one more conflict

* adjust code structure in mmcv.ipu

* adjust ipu code structure in mmcv

* IPUDataloader to IPUDataLoader

* align with mmcv

* adjust according to mmcv

* mmcv code structre fixed

Co-authored-by: hudi <dihu@graphcore.ai>
 2022-04-29 22:22:19 +08:00
mzr1996 5a67bc8a1a [Fix] Fix unexpected modification on `image_demo.py` 2022-04-01 16:50:23 +08:00
Ezra-Yu 922a31a68e
[Feature] ADD RepMLP Backbone and checkpoints. (#709) 
* add repmlp

* refactor backbone code

* add checkpoint, readme, unit tests

* unchange demo

* improve docstring.

* update pic

* refactor patch embed

* refactor unit tests

* fix lint

* update tools

* update tools

* Update checkpoint path

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-03-30 20:40:25 +08:00
Ezra-Yu 131d8c6296
[Feature] Add CAM visualization tool (#577) 
* add cam-grad tool

* refactor cam-grad tool

* add docs

* update docs

* Update docs and support Transformer

* remove pictures and use link

* replace example img and finish EN docs

* improve docs

* improve code

* Fix MobileNet V3 configs

* Refactor to support more powerful feature extraction.

* Add unit tests

* Fix unit test

* fix distortion of visualization exapmles in docs

* fix distortion

* fix distortion

* fix distortion

* merge master

* merge fix conficts

* Imporve the tool

* Support use both attribute name and index to get layer

* add default get_target-layers

* add default get_target-layers

* update docs

* update docs

* add additional printt info when not using target-layers

* Imporve docs

* Fix enumerate list.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2021-12-23 18:53:40 +08:00
Ma Zerun f9eb9b409b
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
Lei Yang 24fd4fb627
Visualize results on image demo (#58) 
* visualize results on image demo

* add matplotlib in requirements
 2020-10-10 16:33:27 +08:00
Lei Yang 9547e7b7a5
Add model inference (#16) 
* add model inference on single image

* rm --eval

* revise doc

* add inference tool and demo

* fix linting

* rename inference_image to inference_model

* infer pred_label and pred_score

* fix linting

* add docstr for inference

* add remove_keys

* add doc for inference

* dump results rather than outputs

* add class_names

* add related infer scripts

* add demo image and the first part of colab tutorial

* conduct evaluation in dataset

* return lst in simple_test

* compuate topk accuracy with numpy

* return outputs in test api

* merge inference and evaluation tool

* fix typo

* rm gt_labels in test conifg

* get gt_labels during evaluation

* sperate the ipython notebook to another PR

* return tensor for onnx_export

* detach var in simple_test

* rm inference script

* rm inference script

* construct data dict to replace LoadImage

* print first predicted result if args.out is None

* modify test_pipeline in inference

* refactor class_names of imagenet

* set class_to_idx as a property in base dataset

* output pred_class during inference

* remove unused docstr
 2020-09-30 19:00:20 +08:00