a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests 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 import random
from time import time from time import time
import kornia
import numpy as np import numpy as np
import torch import torch
import torchvision import torchvision
from torch.utils.data import Dataset 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 from kornia import filters
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
@ -235,7 +237,7 @@ class RandomSharedRegionCrop(nn.Module):
should_flip = random.random() < .5 should_flip = random.random() < .5
if should_flip: if should_flip:
should_flip = 1 should_flip = 1
p2_resized = kornia.geometry.transform.hflip(p2_resized) p2_resized = geometry.transform.hflip(p2_resized)
else: else:
should_flip = 0 should_flip = 0