summaryrefslogtreecommitdiff
AgeCommit message (Collapse)Author
2021-02-17Merge branch 'master' into data_parallelJordan Gong
2021-02-17Merge branch 'master' into python3.8Jordan Gong
2021-02-17Fix type hints and add constrains to height and widthJordan Gong
2021-02-17Add new preprocess scriptJordan Gong
2021-02-16Merge branch 'python3.8' into data_parallel_py3.8Jordan Gong
2021-02-16Merge branch 'master' into data_parallelJordan Gong
2021-02-16Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/model.py
2021-02-16Split transform and evaluate methodJordan Gong
2021-02-15Merge branch 'data_parallel' into data_parallel_py3.8Jordan Gong
2021-02-15Add DataParallel support on new codebaseJordan Gong
2021-02-15Merge branch 'python3.8' into data_parallel_py3.8Jordan Gong
# Conflicts: # models/model.py # models/rgb_part_net.py
2021-02-15Add DataParallel support on new codebaseJordan Gong
2021-02-15Revert "Memory usage improvement"Jordan Gong
This reverts commit be508061
2021-02-15Revert "Memory usage improvement"Jordan Gong
This reverts commit be508061
2021-02-14Merge branch 'master' into python3.8Jordan Gong
2021-02-14Memory usage improvementJordan Gong
This update separates input data to two batches, which reduces ~30% memory usage.
2021-02-14Prepare for DataParallelJordan Gong
2021-02-13Sum gallery dimension instead of all dimensionsJordan Gong
2021-02-10Merge branch 'master' into python3.8Jordan Gong
2021-02-10Implement new sampling technique mentioned in GaitPart[1]Jordan Gong
[1]C. Fan et al., “GaitPart: Temporal Part-Based Model for Gait Recognition,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2020, pp. 14225–14233.
2021-02-10Save scheduler state_dictJordan Gong
2021-02-09Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/rgb_part_net.py
2021-02-09Improve performance when disentanglingJordan Gong
This is a HUGE performance optimization, up to 2x faster than before. Mainly because of the replacement of randomized for-loop with randomized tensor.
2021-02-09Some optimizationsJordan Gong
1. Scheduler will decay the learning rate of auto-encoder only 2. Write learning rate history to tensorboard 3. Reduce image log frequency
2021-02-08Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/hpm.py # models/layers.py # models/model.py # models/rgb_part_net.py # utils/configuration.py
2021-02-08Code refactoring, modifications and new featuresJordan Gong
1. Decode features outside of auto-encoder 2. Turn off HPM 1x1 conv by default 3. Change canonical feature map size from `feature_channels * 8 x 4 x 2` to `feature_channels * 2 x 16 x 8` 4. Use mean of canonical embeddings instead of mean of static features 5. Calculate static and dynamic loss separately 6. Calculate mean of parts in triplet loss instead of sum of parts 7. Add switch to log disentangled images 8. Change default configuration
2021-01-23Remove the third term in canonical consistency lossJordan Gong
2021-01-23Add late start support for non-disentangling partsJordan Gong
2021-01-23Transform all frames together in evaluationJordan Gong
2021-01-23Type hint fixesJordan Gong
2021-01-23Merge branch 'master' into python3.8Jordan Gong
2021-01-23Evaluation bug fixes and code reviewJordan Gong
1. Return full cached clip in evaluation 2. Add multi-iter checkpoints support in evaluation 3. Remove duplicated code while transforming
2021-01-22Handle unexpected restore iterJordan Gong
1. Skip finished model before load it 2. Raise error when restore iter is greater than total iter
2021-01-21A type hint fixJordan Gong
2021-01-21Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # utils/configuration.py
2021-01-21Print average losses after 100 itersJordan Gong
2021-01-21Bug fixesJordan Gong
1. Turn off autograd while decoding canonical and pose features 2. Change default batch size to (4, 8)
2021-01-20Bump up version for scikit-learnJordan Gong
2021-01-14Enable optimizer fine tuningJordan Gong
2021-01-14Remove DataParallelJordan Gong
2021-01-14Remove DataParallelJordan Gong
2021-01-13Remove more stuffJordan Gong
2021-01-13Tensorboard typo fixJordan Gong
2021-01-13Add server startup scriptJordan Gong
2021-01-13Merge branch 'master' into python3.8Jordan Gong
2021-01-13Update config file and convert int to str when joiningJordan Gong
2021-01-13Type hint fixesJordan Gong
2021-01-13Merge branch 'master' into python3.8Jordan Gong
# Conflicts: # models/model.py
2021-01-13Add multiple checkpoints for different model and set default config valueJordan Gong
2021-01-12Merge branch 'master' into python3.8Jordan Gong