149 Commits

Author SHA1 Message Date
littletomatodonkey
3b4f5f4dfc
add distillation and fix some apis (#810)
* fix save load and imagenet dataset
* refine trainer
2021-06-10 16:30:05 +08:00
Felix
c13fe19643
Merge branch 'develop_reg' into patch-6 2021-06-07 15:51:12 +08:00
Felix
c4a4f02f10
Create logo_dataset.py 2021-06-07 15:45:53 +08:00
Felix
819f44db91
Update __init__.py 2021-06-07 15:45:02 +08:00
Bin Lu
75b9b6d284
Update __init__.py 2021-06-07 15:42:30 +08:00
Bin Lu
429409954a
Create icartoon_dataset.py 2021-06-07 15:41:26 +08:00
dongshuilong
3ef9993026 fix ReID trainer bugs 2021-06-03 20:14:45 +08:00
dongshuilong
c784209355 fix ReID trainer 2021-06-03 19:24:50 +08:00
weishengyu
587b2de58c dbg 2021-06-03 15:31:18 +08:00
weishengyu
582157f209 modify format and remove loss 2021-06-03 15:23:23 +08:00
weishengyu
51f0b78bd4 move flowers to root 2021-06-03 15:16:05 +08:00
weishengyu
65f0badf14 merge dataset and sampler -> dataloader 2021-06-03 15:14:05 +08:00
Wei Shengyu
96d1feb682
Merge pull request #773 from RainFrost1/develop_reg
Develop reg
2021-06-03 14:06:27 +08:00
dongshuilong
bba0cf8fdb add CompCars train 2021-06-02 20:04:24 +08:00
cuicheng01
1e8a020f2b fix bug of image transforms 2021-06-02 10:57:38 +00:00
cuicheng01
d13cb22378 Add legendary_model's congigs and fix some trainer's bug 2021-06-01 17:24:21 +00:00
littletomatodonkey
0a0e34298b
add support for infer and export (#760)
* add support for infer

* add export model support

* fix yaml

* fix post process name

* fix topk name
2021-06-01 14:37:47 +08:00
Felix
46955a262d
Merge branch 'develop_reg' into develop_reg 2021-05-31 21:57:23 +08:00
Felix
c9690673ca
Update multilabel_dataset.py 2021-05-31 16:49:54 +08:00
Felix
9c11ddca37
Update __init__.py 2021-05-31 16:47:57 +08:00
Felix
b0708e6e9a
Update __init__.py 2021-05-31 16:40:45 +08:00
Felix
771449813c
Update DistributedRandomIdentitySampler.py 2021-05-31 16:01:39 +08:00
Felix
bf8f2bef51
Update __init__.py 2021-05-31 16:00:32 +08:00
Felix
89a4c43187
Update random_erasing.py 2021-05-31 15:59:53 +08:00
Felix
f2c0cee05c
Update randaugment.py 2021-05-31 15:59:08 +08:00
Felix
ea08c644bd
Update operators.py 2021-05-31 15:58:46 +08:00
Felix
289a6fc11a
Update hide_and_seek.py 2021-05-31 15:58:16 +08:00
Felix
f958e2c05e
Update grid.py 2021-05-31 15:57:54 +08:00
Felix
2f84cc3809
Update functional.py 2021-05-31 15:56:42 +08:00
Felix
017d95caf2
Update fmix.py 2021-05-31 15:55:51 +08:00
Felix
f15a478fc3
Update cutout.py 2021-05-31 15:55:28 +08:00
Felix
50f00e8486
Update autoaugment.py 2021-05-31 15:54:20 +08:00
Felix
6db0ae8af4
Update batch_operators.py 2021-05-31 15:53:54 +08:00
Felix
d2630c3a31
Update multilabel_dataset.py 2021-05-31 15:52:33 +08:00
Felix
56b4d8b889
Update multilabel_dataset.py 2021-05-31 15:49:57 +08:00
Felix
15fd4414d1
Update multilabel_dataset.py 2021-05-31 15:49:19 +08:00
Felix
fa8dbc7f18
Update imagenet_dataset.py 2021-05-31 15:46:54 +08:00
Felix
f83f6f2db8
Create common_dataset.py 2021-05-31 15:43:57 +08:00
Felix
dc49f54ef8
Update random_erasing.py 2021-05-31 13:14:07 +08:00
Felix
d250a69ecd
Update randaugment.py 2021-05-31 13:13:39 +08:00
Felix
7dcdd61c76
Update operators.py 2021-05-31 13:13:20 +08:00
Felix
514e3afe76
Update hide_and_seek.py 2021-05-31 13:13:03 +08:00
Felix
20bfca7522
Update grid.py 2021-05-31 13:12:46 +08:00
Felix
bad16c2570
Update fmix.py 2021-05-31 13:12:27 +08:00
Felix
a92060196c
Update cutout.py 2021-05-31 13:12:07 +08:00
Felix
d98311ea40
Update batch_operators.py 2021-05-31 13:11:50 +08:00
Felix
a6144282a1
Update autoaugment.py 2021-05-31 13:11:31 +08:00
Felix
edf49437c6
Update __init__.py 2021-05-31 13:11:10 +08:00
Felix
71d139fcb7
Update random_erasing.py 2021-05-31 13:10:39 +08:00
Felix
7dd79219ea
Update randaugment.py 2021-05-31 13:10:18 +08:00