9 lines
228 B
Python
9 lines
228 B
Python
# Glue layer for transforming whole pictures into 384x384 sequence for encoder
|
|
# input
|
|
import torch
|
|
import torch.nn as nn
|
|
import torch.nn.functional as F
|
|
import torchvision
|
|
from torchvision import transforms
|
|
|
|
import numpy as np
|