Browse Source

Update DenseNet121-ImageNet

master
mbnacwlh3 4 years ago
parent
commit
84ee402204
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      DenseNet121-ImageNet

+ 1
- 1
DenseNet121-ImageNet View File

@@ -5,7 +5,7 @@ import os
os.environ['TL_BACKEND'] = 'tensorflow' os.environ['TL_BACKEND'] = 'tensorflow'
import tensorlayer as tl import tensorlayer as tl
from densenet import densenet
from .densenet import densenet
tl.logging.set_verbosity(tl.logging.DEBUG) tl.logging.set_verbosity(tl.logging.DEBUG)
def load_ImageNet_dataset(shape=(-1, 256, 256, 3), plotable=False): def load_ImageNet_dataset(shape=(-1, 256, 256, 3), plotable=False):


Loading…
Cancel
Save