AI/U-2-Net
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/xuebinqin/U-2-Net synced 2025-08-05 14:20:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

torchvision's models not used

Browse Source
This commit is contained in:
Jason Dsouza 2020-11-24 16:05:49 +05:30 committed by GitHub
parent cc5d70f6e0
commit 6aa8e3cb39
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
model/u2net.py
View File

@ -1,6 +1,5 @@
import torch
import torch.nn as nn
from torchvision import models
import torch.nn.functional as F
class REBNCONV(nn.Module):