Commit Graph

76 Commits (9684500a57c5e90b343e863e2a0292471ecfefb9)
 

Author SHA1 Message Date
liaoxingyu 9684500a57 chagne arch
1. change dataset show to trainset show and testset show seperately
2. add cls layer to easily plug in circle loss and arcface
2020-04-19 12:54:01 +08:00
liaoxingyu be9faa5605 update focal loss
update dataset info display
update seperate lr
update adaptive label smooth regularization
2020-04-17 13:46:10 +08:00
liaoxingyu 9cf222e093 refactor bn_no_bias 2020-04-08 21:04:09 +08:00
liaoxingyu 4d2fa28dbb update freeze layer
update preciseBN
update circle loss with metric learning and cross entropy loss form
update loss call methods
2020-04-06 23:34:27 +08:00
liaoxingyu 6a8961ce48 1. upload circle loss and arcface
2. finish freeze training
3. update augmix data augmentation
2020-04-05 23:54:26 +08:00
liaoxingyu c6e0176c53 Upload demo.py and example 2020-04-03 15:07:27 +08:00
liaoxingyu 3f3975e372 Update performance 2020-03-31 14:37:24 +08:00
liaoxingyu eacee874aa fix merge 2020-03-25 11:06:39 +08:00
liaoxingyu 91dc9bc71f Merge branch 'master' of github.com:L1aoXingyu/fast-reid
 Conflicts:
	fastreid/config/defaults.py
	fastreid/layers/gem_pool.py
	fastreid/modeling/backbones/resnet.py
	fastreid/modeling/heads/__init__.py
	fastreid/modeling/heads/build.py
	fastreid/modeling/losses/build.py
	fastreid/modeling/meta_arch/__init__.py
	fastreid/modeling/meta_arch/abd_network.py
	fastreid/modeling/meta_arch/baseline.py
	fastreid/modeling/meta_arch/bdb_network.py
	fastreid/modeling/meta_arch/mf_network.py
	projects/StrongBaseline/configs/Base-Strongbaseline.yml
	projects/StrongBaseline/configs/baseline_dukemtmc.yml
	projects/StrongBaseline/train_net.py
2020-03-25 11:05:28 +08:00
liaoxingyu 23bedfce12 update version0.2 code 2020-03-25 10:58:26 +08:00
L1aoXingyu b1058118ca update BDB-net code
update MF-net code
2020-03-19 12:23:41 +08:00
L1aoXingyu b2076fa39d 1. Update cython code for computing mINP 2020-03-16 15:29:01 +08:00
L1aoXingyu acf363c181 1. Change loss function as a build-in attributes of heads
2. Update agw and bagtricks result
2020-03-16 15:23:09 +08:00
L1aoXingyu bab602dfd2 Fix minor bug in build criterion, it will replace by multiple call
Refactor resnet pretrain
2020-02-28 21:20:41 +08:00
L1aoXingyu b020c7f0ae Fix data prefetcher minor bug 2020-02-27 12:16:57 +08:00
L1aoXingyu 12957f66aa Change architecture:
1. delete redundant preprocess
2. add data prefetcher to accelerate data loading
3. fix minor bug of triplet sampler when only one image for one id
2020-02-18 21:01:23 +08:00
L1aoXingyu e01d9b241f Update AGW baseline result 2020-02-13 20:37:08 +08:00
L1aoXingyu 327d74ffbb Update strong baseline result
Change data sampler
2020-02-13 00:19:15 +08:00
L1aoXingyu a2f69d0537 Update StrongBaseline results for market1501 and dukemtmc 2020-02-11 22:38:40 +08:00
L1aoXingyu 8a9c0ccfad Finish first version for fastreid 2020-02-10 22:13:04 +08:00
L1aoXingyu db6ed12b14 Update sampler code 2020-02-10 07:38:56 +08:00
liaoxingyu 71950d2c09 1. Fix evaluation code
2. Finish multi-dataset evaluation
3. Decouple image preprocess and output postprocess with model forward for DataParallel training
4. Finish build backbone registry
5. Fix dataset sampler
2020-01-21 20:24:26 +08:00
liaoxingyu b761b656f3 Finish basic training loop and evaluation results 2020-01-20 21:33:37 +08:00
liaoxingyu 315ef25801 Update code
Initialize code
2020-01-17 18:55:57 +08:00
liaoxingyu da27da41b5
update readme
update
2019-12-08 15:56:01 +08:00
liaoxingyu 48232b18cd
Update README.md 2019-11-14 13:59:28 +08:00
liaoxingyu 9a3d365e9b <feature> fix bug and remove other dependency
1. remove pytorch-lightning dependency, and now it's pure pytorch code.
2. add prefetcher dataloader to accelerate data loading
3. add ibn and new training engine
4. add multi-dataset support
2019-09-20 18:48:15 +08:00
liaoxingyu 15a911879a <feature> fix readme 2019-09-20 15:39:20 +08:00
L1aoXingyu 9f88ced0ab <refactor> change code to pytorch-lightning
1. Refactor code, and change code style to pytorch-lightning due to the flexibility.
2. Fix sampler lens bug.
2019-09-04 00:05:03 +08:00
L1aoXingyu a9d1b5080e Merge remote-tracking branch 'origin/master'
# Conflicts:
#	README.md
#	modeling/backbones/resnet.py
2019-08-26 23:11:32 +08:00
L1aoXingyu 8e2a4da6f9 refactor code 2019-08-26 23:10:08 +08:00
liaoxingyu 4814b8d602 1. add label smooth
2. update experiment
2019-08-26 11:31:37 +08:00
liaoxingyu c54ae788ff update readme 2019-08-23 23:36:19 +08:00
liaoxingyu 8575b2ad77 update gcnet result 2019-08-23 07:49:03 +08:00
liaoxingyu d731ed3031 finish actmap code 2019-08-21 13:52:41 +08:00
liaoxingyu 3d5f7d24aa 1. fix minor bug
2. update experiment results in readme
2019-08-21 09:35:34 +08:00
liaoxingyu aa4c8eee2c update code 2019-08-20 09:36:47 +08:00
liaoxingyu fd751ef197 update code 2019-08-15 13:28:03 +08:00
liaoxingyu 8ac3971fa5 update readme 2019-08-15 08:59:24 +08:00
liaoxingyu c18402624a
strong baseline ++ (#25)
1. transfer from ignite to fastai
2. multi-data training
3. stronger baseline with ibn
4. cross dataset evaluation
2019-08-14 14:55:52 +08:00
liaoxingyu 5fd586492f release ready 2019-08-14 14:50:44 +08:00
liaoxingyu 88dfd71a76 Finish ibn code 2019-08-13 13:52:25 +08:00
liaoxingyu 872e370142 add new feature
Finish interpreter
2019-08-07 16:54:50 +08:00
liaoxingyu 710d40aec0 interp
Finish visulization code and interpretation
2019-08-02 19:30:16 +08:00
liaoxingyu 48518552cf update
Add cuhk03
2019-08-01 15:49:31 +08:00
liaoxingyu 369325906f update
Add multi-data training
Add beijing data testing
Fix testing bug
2019-07-31 21:11:54 +08:00
liaoxingyu 08d105560c code
Finish multi-dataset training
2019-07-30 18:56:43 +08:00
liaoxingyu defc190f6b code
finish training on market
2019-07-29 20:57:19 +08:00
liaoxingyu d4165c09aa code
Finish training on market
2019-07-29 20:57:06 +08:00
liaoxingyu f7f26ae40a Fast fix 2019-07-23 09:46:37 +08:00