mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Another fix

Browse Source
This commit is contained in:
James Betker 2022-04-08 09:41:18 -06:00
parent 423293e518
commit 628569af7b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
codes/data/images/byol_attachment.py
View File

@ -1,11 +1,13 @@
import random
from time import time
import kornia
import numpy as np
import torch
import torchvision
from torch.utils.data import Dataset
from kornia import augmentation as augs, kornia, Resample
from kornia import augmentation as augs, geometry
from kornia import filters
import torch.nn as nn
import torch.nn.functional as F
@ -235,7 +237,7 @@ class RandomSharedRegionCrop(nn.Module):
should_flip = random.random() < .5
if should_flip:
should_flip = 1
p2_resized = kornia.geometry.transform.hflip(p2_resized)
p2_resized = geometry.transform.hflip(p2_resized)
else:
should_flip = 0