claussteinmassl 1891671340 Refactor calls to cuda to be more generic, so it works on cuda, mps and cpu, depending on the available device.

2024-08-06 21:11:26 +02:00

..

__init__.py

main

2024-08-03 15:41:49 +09:00

appearance_feature_extractor.py

main

2024-08-03 15:41:49 +09:00

convnextv2.py

main

2024-08-03 15:41:49 +09:00

dense_motion.py

Refactor calls to cuda to be more generic, so it works on cuda, mps and cpu, depending on the available device.

2024-08-06 21:11:26 +02:00

motion_extractor.py

main

2024-08-03 15:41:49 +09:00

spade_generator.py

main

2024-08-03 15:41:49 +09:00

stitching_retargeting_network.py

main

2024-08-03 15:41:49 +09:00

util.py

main

2024-08-03 15:41:49 +09:00

warping_network.py

main

2024-08-03 15:41:49 +09:00