A simple 4D convolution layer and 4D cost volume layer in CUDA with Pytorch interface.
Compile and test
cd cpp
python setup.py install
python test.py
cd ..
cd cuda
python setup.py install
python test.py
-
Notifications
You must be signed in to change notification settings - Fork 0
yaolubrain/Conv4D
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published