Commit Graph

244 Commits (ccdafe32cae926943d18a32958e94d314c00d0cc)

Author SHA1 Message Date
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
HydrogenSulfate 84870b771d add assert for DistributedRandomIdentitySampler and refine docstring for DistributedRandomIdentitySampler&Pksampler 2022-08-18 20:16:55 +08:00
Yang Nie 3248cc5819 update `DecodeImage` 2022-07-09 23:30:13 +08:00