Update config.py

master
SaiYiKi 2019-08-06 17:57:28 +08:00 committed by GitHub
parent b946c9a7ab
commit eae9fcc510
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -67,7 +67,7 @@ class LoadConfig(object):
self.rawdata_root = './../FGVC_product/data'
self.anno_root = './../FGVC_product/anno'
self.numcls = 2019
if args.dataset == 'CUB':
elif args.dataset == 'CUB':
self.dataset = args.dataset
self.rawdata_root = './dataset/CUB_200_2011/data'
self.anno_root = './dataset/CUB_200_2011/anno'