246 Commits

Author SHA1 Message Date
HydrogenSulfate
7e12c73e0a polish progressive training code 2023-01-31 21:15:22 +08:00
HydrogenSulfate
3a8b56801d feat(model): add EfficientNetV2 code and fix AttrDict BUG 2023-01-31 21:15:22 +08:00
gaotingquan
f82871b1f8 add copyright & reference 2023-01-06 20:29:07 +08:00
gaotingquan
e41f2a8fee support PCALighting 2023-01-06 20:29:07 +08:00
zh-hike
f8978a2f0c 修改cifar100参数配置 2023-01-04 13:42:00 +08:00
zh-hike
f073e97d22 增加cifar100参数yaml 2023-01-04 13:42:00 +08:00
zh-hike
009f347d64 修改空格等代码规范 2023-01-04 13:42:00 +08:00
zh-hike
f83ff59c97 增加代码规范,删除一些无用的function 2023-01-04 13:42:00 +08:00
zh-hike
692b8d8c1f 增加data的代码复用及RecModel的修改,代码已跑通 2023-01-04 13:42:00 +08:00
zh-hike
7823f340b1 cifar10_4000跑出论文精度代码 2023-01-04 13:42:00 +08:00
zhangyubo0722
169682002d Some modifications 2022-12-28 16:01:54 +08:00
zhangyubo0722
7badd5bf6c add metrics and postprocess 2022-12-28 16:01:54 +08:00
zengshao0622
885e1bc05f fix numpy int bug 2022-12-22 14:19:15 +08:00
zengshao0622
7cdae10bcf fix pksampler prob list shuffle bug 2022-12-22 12:01:05 +08:00
HydrogenSulfate
4060957669 fix(dali): add shuffle to prob_list in ExternalSource_RandomIdentity 2022-12-13 14:27:15 +08:00
HydrogenSulfate
b22752e2ea refactor DALI 2022-12-13 14:27:15 +08:00
HydrogenSulfate
6cc6540ca5 add different seed for workers and replicas 2022-11-24 11:51:13 +08:00
HydrogenSulfate
b542416d67 add random shuffle in PKSampler 2022-11-24 11:51:13 +08:00
HydrogenSulfate
fa52acd614 remove redundant import in pksampler 2022-11-22 16:47:47 +08:00
HydrogenSulfate
c26adddf33 remove redundant code in pksampler 2022-11-22 16:47:47 +08:00
HydrogenSulfate
7ff257ea69 fix random seed bug for pksampler in DDP 2022-11-22 16:47:47 +08:00
dongshuilong
944763d7a5 add fixmatch 2022-10-25 12:04:22 +08:00
HydrogenSulfate
a14df4ac52 fix tensor conversion in static mode with dali loader 2022-10-20 12:03:03 +00:00
HydrogenSulfate
09200a31f4 remove redundant code, fix bugs in lr.step, merge GoodsDataset into Vehicle 2022-10-19 07:18:21 +00:00
HydrogenSulfate
2cfd8dd828 add xbm 2022-10-17 15:45:45 +08:00
gaotingquan
b83da15d27 support set shuffle by config 2022-10-17 11:24:57 +08:00
HydrogenSulfate
61b4153907 add batch Tensor collate to simplify dali code in train/eval/retrival code 2022-10-13 12:05:50 +08:00
cuicheng01
c87db7b9e3 update blur code 2022-10-11 13:46:27 +00:00
cuicheng01
083205273d add clarity_assessment code 2022-09-29 06:25:22 +00:00
gaotingquan
d670f2e497 rename config file ImageOrientation to PPLCNet_x1_0 2022-09-28 20:01:48 +08:00
gaotingquan
bf91fbfbe4 add warning msg 2022-09-28 20:01:48 +08:00
gaotingquan
75e1291528 debug 2022-09-28 20:01:48 +08:00
gaotingquan
19840cb046 refactor: to be pythonic 2022-09-28 20:01:48 +08:00
gaotingquan
e823f17853 feat: support training image orientation model
add the config of orientation model
add the preprocess op RandomRot90 that can rotate the img and return the orientation
add the CustomLabelDataset that support getting label by preprocess
refactor some preprocess ops to support dict parameter and return dict
2022-09-28 20:01:48 +08:00
cuicheng01
74eb371dd6 add table_attribute docs 2022-09-20 04:37:52 +00:00
Tingquan Gao
9344351493
Merge pull request #2219 from cuicheng01/fix_multilabel
fix multilabel
2022-09-15 10:54:28 +08:00
HydrogenSulfate
42758f5463 fix benchmark train's config for GeneralRecognitionV2, and polish TIPC docs 2022-09-14 08:54:22 +00:00
cuicheng01
0b9f62262a
Merge branch 'develop' into fix_multilabel 2022-09-14 14:41:32 +08:00
Tingquan Gao
2ed68d5dba
Merge pull request #2264 from cuicheng01/add_table_attribute
add table_attribute_code
2022-09-14 14:17:11 +08:00
zengshao0622
7b50ce6585 merge CAE 2022-09-08 08:11:25 +00:00
cuicheng01
f95343590b fix format 2022-09-07 10:04:11 +00:00
cuicheng01
18acc25f18 add table_attribute_code 2022-09-07 09:35:51 +00:00
gaotingquan
3f8f981423 fix: support specify DALI threads num 2022-09-01 20:21:43 +08:00
cuicheng01
85bf8683cb Support for parsing label files 2022-08-29 05:49:15 +00:00
HydrogenSulfate
73e1d36697 refine doc-string for ImageNetDataset 2022-08-25 14:40:19 +08:00
HydrogenSulfate
877c8c53be correct config yaml, pp_lcnetv2_variant.py and log in&out channels in pp_lcnet_v2.py 2022-08-24 07:25:30 +00:00
HydrogenSulfate
5a4874079d fix some bugs 2022-08-23 10:59:30 +00:00
HydrogenSulfate
1b5e00e82a add PP-ShiTuV2 code 2022-08-23 07:54:58 +00:00
cuicheng01
eade1b7232 fix multilabel 2022-08-22 06:40:26 +00:00
HydrogenSulfate
a646adf754 refine docstring 2022-08-18 20:20:07 +08:00