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-03 20:20:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
232 Commits 4 Branches 0 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
Xuebin Qin
3ef086adfb
Merge pull request #8 from szerintedmi/resolve_0-dim_tensor_warnings 
Resolve invalid index of a 0-dim tensor warnings
 2021-01-23 17:24:56 -07:00
Diego Porres
9fe1d94ff2
Update u2net_test.py 
Make code OS-independent.
 2020-05-13 09:05:32 +02:00
szerintedmi
4031385d80 map location to work on torch < 4.0.1 2020-05-11 18:10:21 +01:00
szerintedmi
1b247c5419 load model to cpu when no cuda 2020-05-11 12:01:26 +01:00
Nathan Qin
e660dc6c5a new 2020-05-09 15:32:41 -06:00