File tree 1 file changed +43
-0
lines changed
1 file changed +43
-0
lines changed Original file line number Diff line number Diff line change
1
+ from pytti .image_models .differentiable_image import DifferentiableImage
2
+ from pytti .image_models .ema import EMAImage
3
+ from pytti .image_models .pixel import PixelImage
4
+ from pytti .image_models .rgb_image import RGBImage
5
+ from pytti .image_models .vqgan import VQGANImage
6
+ from pytti .image_models .deep_image_prior import DeepImagePrior
7
+
8
+ ### DIP ###
9
+
10
+
11
+ def test_dip_init ():
12
+ obj = DeepImagePrior (512 , 512 )
13
+ assert obj
14
+
15
+
16
+ def test_dip_update ():
17
+ obj = DeepImagePrior (512 , 512 )
18
+ obj .update ()
19
+
20
+
21
+ def test_dip_forward ():
22
+ obj = DeepImagePrior (512 , 512 )
23
+ obj .forward ()
24
+
25
+
26
+ def test_dip_decode_training_tensor ():
27
+ obj = DeepImagePrior (512 , 512 )
28
+ obj .decode_training_tensor ()
29
+
30
+
31
+ def test_dip_decode_tensor ():
32
+ obj = DeepImagePrior (512 , 512 )
33
+ obj .decode_tensor ()
34
+
35
+
36
+ def test_dip_clone ():
37
+ obj = DeepImagePrior (512 , 512 )
38
+ obj .clone ()
39
+
40
+
41
+ def test_dip_get_latent_tensor ():
42
+ obj = DeepImagePrior (10 , 10 )
43
+ obj .get_latent_tensor ()
You can’t perform that action at this time.
0 commit comments