fix the bug of reading images

xiaomengyc · Aug 15, 2018 · 8b59d00 · 8b59d00
1 parent 5491d7c
commit 8b59d00
Show file tree

Hide file tree

Showing 17 changed files with 6 additions and 5 deletions.
diff --git a/exper/my_optim.pyc b/exper/my_optim.pyc
diff --git a/exper/train_frame.py b/exper/train_frame.py
@@ -40,7 +40,7 @@ def get_arguments():
     parser = argparse.ArgumentParser(description='ACoL')
     parser.add_argument("--root_dir", type=str, default=ROOT_DIR,
                         help='Root dir for the project')
-    parser.add_argument("--img_dir", type=str, default=IMG_DIR,
+    parser.add_argument("--img_dir", type=str, default='',
                         help='Directory of training images')
     parser.add_argument("--train_list", type=str,
                         default=train_list)

diff --git a/models/__init__.pyc b/models/__init__.pyc
diff --git a/models/vgg/__init__.pyc b/models/vgg/__init__.pyc
diff --git a/models/vgg/vgg_v0.pyc b/models/vgg/vgg_v0.pyc
diff --git a/models/vgg/vgg_v1.pyc b/models/vgg/vgg_v1.pyc
diff --git a/scripts/train_vgg_cub.sh b/scripts/train_vgg_cub.sh
@@ -6,12 +6,12 @@ cd ../exper/
 THRESHOLD=0.6
 
 
-CUDA_VISIBLE_DEVICES=0 python train_frame.py --arch=vgg_v1 --epoch=60 --lr=0.0001 --batch_size=20 --num_gpu=1 --dataset=cub  \
+CUDA_VISIBLE_DEVICES=0,1 python train_frame.py --arch=vgg_v1 --epoch=60 --lr=0.0001 --batch_size=20 --num_gpu=2 --dataset=cub  \
 	--disp_interval=40 \
-	--num_classes=1000 \
+	--num_classes=200 \
 	--threshold=${THRESHOLD} \
 	--num_workers=6 \
-	--img_dir=../data/CUB_200_2011/images/ \
 	--train_list=../datalist/CUB/train_list.txt \
+	--img_dir=../data/CUB_200_2011/images \
 	--snapshot_dir=../snapshots/vgg_imagenet_v1/  \
 	--restore_from=/home/xiaolin/.torch/models/vgg16-397923af.pth \
diff --git a/utils/LoadData.pyc b/utils/LoadData.pyc
diff --git a/utils/Metrics.pyc b/utils/Metrics.pyc
diff --git a/utils/Restore.pyc b/utils/Restore.pyc
diff --git a/utils/__init__.pyc b/utils/__init__.pyc
diff --git a/utils/avgMeter.pyc b/utils/avgMeter.pyc
diff --git a/utils/mydataset.py b/utils/mydataset.py
@@ -33,7 +33,8 @@ def __len__(self):
         return len(self.image_list)
 
     def __getitem__(self, idx):
-        img_name = os.path.join(self.root_dir, self.image_list[idx])
+        #img_name = os.path.join(self.root_dir, self.image_list[idx])
+        img_name =  self.image_list[idx]
         image = Image.open(img_name).convert('RGB')
 
         if self.transform is not None:

diff --git a/utils/mydataset.pyc b/utils/mydataset.pyc
diff --git a/utils/transforms/__init__.pyc b/utils/transforms/__init__.pyc
diff --git a/utils/transforms/functional.pyc b/utils/transforms/functional.pyc
diff --git a/utils/transforms/transforms.pyc b/utils/transforms/transforms.pyc