()
| 591 | |
| 592 | |
| 593 | def celeb(): |
| 594 | X = util.get_celeb() |
| 595 | # just loads a list of filenames, we will load them in dynamically |
| 596 | # because there are many |
| 597 | dim = 64 |
| 598 | colors = 3 |
| 599 | |
| 600 | # for celeb |
| 601 | d_sizes = { |
| 602 | 'conv_layers': [ |
| 603 | (64, 5, 2, False), |
| 604 | (128, 5, 2, True), |
| 605 | (256, 5, 2, True), |
| 606 | (512, 5, 2, True) |
| 607 | ], |
| 608 | 'dense_layers': [], |
| 609 | } |
| 610 | g_sizes = { |
| 611 | 'z': 100, |
| 612 | 'projection': 512, |
| 613 | 'bn_after_project': True, |
| 614 | 'conv_layers': [ |
| 615 | (256, 5, 2, True), |
| 616 | (128, 5, 2, True), |
| 617 | (64, 5, 2, True), |
| 618 | (colors, 5, 2, False) |
| 619 | ], |
| 620 | 'dense_layers': [], |
| 621 | 'output_activation': T.tanh, |
| 622 | } |
| 623 | |
| 624 | # setup gan |
| 625 | # note: assume square images, so only need 1 dim |
| 626 | gan = DCGAN(dim, colors, d_sizes, g_sizes) |
| 627 | gan.fit(X) |
| 628 | |
| 629 | |
| 630 | def mnist(): |
no test coverage detected