u2net.py

import torch.nn as nn

class U2Net(nn.Module):
    def __init__(self):
        super(U2Net, self).__init__()
        pass