Open guiyuliu opened 4 years ago
maybe it should be changed to this
in line 38 in func def download_imagenet(self): if self.backbone == 'vgg16': backbone_name = 'vgg16' resource = keras.applications.vgg16.vgg16.WEIGHTS_PATH_NO_TOP checksum = '6d6bbae143d832006294945121d1f1fc' elif 'vgg-max' in self.backbone: backbone_name = 'vgg16' resource = keras.applications.vgg16.vgg16.WEIGHTS_PATH_NO_TOP checksum = '6d6bbae143d832006294945121d1f1fc' elif self.backbone == 'vgg19': backbone_name = 'vgg19' resource = keras.applications.vgg19.vgg19.WEIGHTS_PATH_NO_TOP checksum = '253f8cb515780f3b799900260a226db6' else: raise ValueError("Backbone '{}' not recognized.".format(self.backbone)) return get_file('{}_weights_tf_dim_ordering_tf_kernels_notop.h5'.format(backbone_name))
in line 38 in func def download_imagenet(self): if self.backbone == 'vgg16': backbone_name = 'vgg16' resource = keras.applications.vgg16.vgg16.WEIGHTS_PATH_NO_TOP checksum = '6d6bbae143d832006294945121d1f1fc' elif 'vgg-max' in self.backbone: backbone_name = 'vgg16' resource = keras.applications.vgg16.vgg16.WEIGHTS_PATH_NO_TOP checksum = '6d6bbae143d832006294945121d1f1fc' elif self.backbone == 'vgg19': backbone_name = 'vgg19' resource = keras.applications.vgg19.vgg19.WEIGHTS_PATH_NO_TOP checksum = '253f8cb515780f3b799900260a226db6' else: raise ValueError("Backbone '{}' not recognized.".format(self.backbone)) return get_file('{}_weights_tf_dim_ordering_tf_kernels_notop.h5'.format(backbone_name))