search

mapooon / SelfBlendedImages

[CVPR 2022 Oral] Detecting Deepfakes with Self-Blended Images https://arxiv.org/abs/2204.08376
Other
196 stars 34 forks source link

About train_sbi.py #23

Closed Olive0407 closed 9 months ago

Olive0407 commented 1 year ago 
    In train_sbi.py:

for step,data in enumerate(tqdm(train_loader)): img=data['img'].to(device, non_blocking=True).float() target=data['label'].to(device, non_blocking=True).long() but in SBI_dataset : only return img_f,img_r. there is no label, could someone solve my problem.

mapooon commented 1 year ago

Labels are generated in collate_fn() of SBI_Dataset.