-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathtutorial.py
More file actions
executable file
·30 lines (25 loc) · 801 Bytes
/
tutorial.py
File metadata and controls
executable file
·30 lines (25 loc) · 801 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import cv2
import numpy as np
# from nodeflux.face_detection.retinaface import RetinaFace
import time
import pdb
# from nodeflux.face_spoof.preprocessing.transform import center_crop
# from nodeflux.face_spoof.feathernet import FeatherNetA
import matplotlib.pyplot as plt
from torch.utils.data import Dataset, DataLoader
from pathlib import Path
from torchvision import transforms
from torchetl.etl import TransformAndLoad
import pdb
import pandas as pd
import pandas as pd
# pdb.set_trace()
parent_directory = Path('data') / 'imdb'
test_dataset_csv = parent_directory / 'imdb_even_cleaner_with_bbox.csv'
df = pd.read_csv(test_dataset_csv)
# pdb.set_trace()
testing_pipeline = transforms.Compose([
transforms.ToPILImage(),
transforms.RandomResizedCrop(224),
transforms.ToTensor(),
])