summaryrefslogtreecommitdiff
path: root/utils/dataset.py
AgeCommit message (Collapse)Author
2021-03-10Bug fixesJordan Gong
1. Resolve reference problems when parsing dataset selectors 2. Transform gallery using different models
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-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-12Some type hint fixesJordan Gong
2020-12-31Make super class constructor revoke consistentJordan Gong
2020-12-29Add type hint for new label (numpy.int64)Jordan Gong
2020-12-29Encode class names to label and some access improvementJordan Gong
1. Encode class names using LabelEncoder from sklearn 2. Remove unneeded class variables 3. Protect some variables from being accessed in userspace
2020-12-27Add dataset selector to config type hint, change ClipLabels typo to ClipViewsJordan Gong
2020-12-27Adopt type hinting generics in standard collections (PEP 585)Jordan Gong
2020-12-26Combine transformed height and width to `frame_size`Jordan Gong
2020-12-21Change image loading techniqueJordan Gong
1. Use Pillow.Image.open instead of torchvision.io.read_image to read image 2. Transforming PIL images instead of tensors which performs better and device option is removed 3. Images are normalized now
2020-12-191. Delete unused transform functionJordan Gong
2. Reorganize the initialization cache dicts
2020-12-19Fix indexing error when no clip to be discardJordan Gong
2020-12-19Add cache switch, allowing load all data into RAM before samplingJordan Gong
2020-12-18Implement CASIA-B datasetJordan Gong