VDataset
Description
Load video datasets to PyTorch DataLoader. (Custom Video Data set for PyTorch DataLoader)
Required Libraries
- torch
- torchvision
- Pillow
- pandas
Arguments for constructor
| Argument | Type | Required | Description |
|---|---|---|---|
| csv_file | str | True | Path to .csv file |
| root_dir | str | True | Root Directory of the video dataset |
| file_format | str | False | File type of the frame images (ex: .jpg, .jpeg, .png) |
| id_col_name | str | False | Column name, where id/name of the video on the .csv file |
| label_col_name | str | False | Column name, where label is on the .csv file |
| frames_limit_mode | str/None | False | Mode of the frame count detection ("manual", "csv" or else it auto detects all the frames available) |
| frames_limit | int | False | Number of frames in a video (required if frames_count_mode set to "manual") |
| frames_limit_col_name | str | False | Column name, where label is on the .csv file (required if frames_count_mode set to "csv") |
| frames_resize | tuple/None | False | Resize the frames (Also this can be done on using transform too) |
Usage
from torch.utils.data import DataLoader
from torchvision import transforms
transforms = transforms.Compose([transforms.Resize((100, 100)),
transforms.ToTensor()])
vdataset = VDataset(csv_file='path-to-csv-file.csv', root_dir='path-to-video-dir', transform=transforms)
dataloader = DataLoader(vdataset, batch_size=64) # use in DataLoader
for image, label in dataloader: # Do what do you want in dataset
print(image, label)
break
1 maintainer
Metadata
1 maintainer
Release files for vdataset 0.0.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| vdataset-0.0.1.tar.gz | 3.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| vdataset-0.0.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 7.7 kB
Release files / vdataset-0.0.1.tar.gz
| Download URL | vdataset-0.0.1.tar.gz |
|---|---|
| Size | 3.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
5185b4f7baa1edf8d7a7c6b4f566fadd30585ac813b41a37d52a19fafe146020
|
|
BLAKE2b-256 checksum How to use checksums |
9dc97070da6e0c0a9fe69fdb9c153ca105be606e92d46be7e3cb29e35dc61b69
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.4.2 importlib_metadata/3.10.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.8.8
|
Release files / vdataset-0.0.1-py3-none-any.whl
| Download URL | vdataset-0.0.1-py3-none-any.whl |
|---|---|
| Size | 4.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
da3e353513f6c00cfda9b05eee66584e729a444597f01b452359ebdd72b9da35
|
|
BLAKE2b-256 checksum How to use checksums |
8cfdbe3333a866d7178587caf173991bb52b08dddc43d6d22522ead76446e778
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.4.2 importlib_metadata/3.10.0 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.59.0 CPython/3.8.8
|
