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

13 Commits (c992e246175777b884ece1838e2dbb610c5becc0)

Author SHA1 Message Date
Ezra-Yu a9057e88c4 Add multi label metrics 2022-07-18 11:11:13 +08:00
mzr1996 dd660ed99e [Improve] Clean useless code and reduce unit tests memory usage. 2022-07-18 11:11:13 +08:00
mzr1996 6ad75f0076 [Feature] Add metrics for single-label classification. 2022-07-18 10:53:28 +08:00
HumberMe 1214df083d
[Enhance] Use PyTorch official `one_hot` to implement `convert_to_one_hot`. (#696) 
* some change to mmcls/models/losses/utils.py:convert_to_one_hot()

* fixed problem: line too long

* fixed wrong output shape

* fixed lint PEP8 E128

* fix lint

* fix lint

* add unit tests

Co-authored-by: Ezra-Yu <1105212286@qq.com>
 2022-02-28 10:20:31 +08:00
Yusuke Minami e694269c59
[Enhance] Support single-label, softmax, custom eps by asymmetric loss (#609) 
Co-authored-by: Minyus <Minyus@users.noreply.github.com>
 2022-01-21 11:35:14 +08:00
Ma Zerun e3cf1881bb
[Enhance] Imporve efficiency of precision, recall, f1_score and support. (#595) 
* [Enhance] Imporve efficiency of precision, recall, f1_score and support.

* Fix bugs

* Use np.maximum since torch doesn't have maximum before torch 1.7

* Fix bug
 2021-12-13 17:24:59 +08:00
Ma Zerun 188aa6ed5d
[Enhance] Improve accuracy calculation performance. (#592) 
* Imporve accuracy calculate performance.

* Add unit tests for accuracy

* Reuse state_inds
 2021-12-09 14:06:08 +08:00
juanjompz 715a80b09c
[Enhance] Focal loss for single label tasks (#548) 
* Added focal loss support to single label tasks.

* Updated focal loss docstring
 2021-11-26 12:36:49 +08:00
Ezra-Yu 87c67a6a79
[Feature] Support positive weights in BCE. (#516) 
* add pos-weight in BCE

* mv pos-weight to kwargs
 2021-11-10 17:14:12 +08:00
Zhicheng Chen 6d6ce215a2
[Feature] Add seesaw loss. (#500) 
* migrate seesaw loss from mmdet

* add assertion and doc string fixes

* add error information

* docstring fixes

* minor doc update
 2021-10-27 10:11:59 +08:00
Ezra-Yu 192b79eea0
[Feature] Support classwise weight in losses (#388) 
* Add classwise weight in losses:CE,BCE,softBCE

* Update unit test

* rm some extra code

* rm some extra code

* fix broadcast

* fix broadcast

* update unit tests

* use new_tensor

* fix lint
 2021-08-31 10:44:12 +08:00
Ma Zerun f9eb9b409b
[Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
Ma Zerun 1a7cebe4b9
[Refactor] Refactor unittest (#321) 
* Refactor unit tests folder structure.

* Remove label smooth and Vit test in `test_classifiers.py`

* Rename test_utils in dataset to test_dataset_utils

* Split test_models/test_utils/test_utils.py to multiple sub files.

* Add unit tests of classifiers and heads

* Use patch context manager.

* Add unit test of `is_tracing`, and add warning in `is_tracing` if torch
verison is smaller than 1.6.0
 2021-07-08 22:49:05 +08:00