fix data transform crop x y inversed bug

Author: huochaitiantang

core/data.py

@@ -39,7 +39,7 @@ def __call__(self, img, alpha, fg, bg, trimap, crop_h, crop_w):
         cropx, cropy = 0, 0
         if len(target[0]) > 0:
             rand_ind = np.random.randint(len(target[0]), size = 1)[0]
-            cropx, cropy = target[0][rand_ind], target[1][rand_ind]
+            cropy, cropx = target[0][rand_ind], target[1][rand_ind]
             cropx = min(max(cropx, 0), w - crop_w)
             cropy = min(max(cropy, 0), h - crop_h)
 

core/train.py

@@ -339,7 +339,7 @@ def main():
 
     # training
     for epoch in range(start_epoch, args.nEpochs + 1):
-        #train(args, model, optimizer, train_loader, epoch)
+        train(args, model, optimizer, train_loader, epoch)
         if epoch > 0 and epoch % args.ckptSaveFreq == 0:
             checkpoint(epoch, args.saveDir, model)
         if epoch > 0 and args.testFreq > 0 and epoch % args.testFreq == 0:

deploy.sh

@@ -1,17 +1,15 @@
 #/bin/bash
 
-DESKTOP=/home/liuliang/Desktop
-MODEL_ROOT=$DESKTOP/pytorch-deep-image-matting-pure
-DATA_ROOT=$DESKTOP/dataset/matting/dataset_deep_image_matting/Test_set
+TEST_DATA_ROOT=/data/datasets/matting/Combined_Dataset/Test_set/comp
 
 python core/deploy.py \
     --size_h=320 \
     --size_w=320 \
-    --imgDir=$DATA_ROOT/comp/image \
-    --trimapDir=$DATA_ROOT/comp/trimap \
-    --alphaDir=$DATA_ROOT/comp/alpha \
-    --saveDir=$MODEL_ROOT/result/batch1_stage1_cktp_e22 \
-    --resume=$MODEL_ROOT/model/batch1_stage0_ckpt_e22.pth \
+    --imgDir=$TEST_DATA_ROOT/image \
+    --trimapDir=$TEST_DATA_ROOT/trimap \
+    --alphaDir=$TEST_DATA_ROOT/alpha \
+    --saveDir=result/stage0 \
+    --resume=model/stage0/ckpt_e19.pth \
     --cuda \
     --stage=0 \
     --crop_or_resize=whole \

train.sh

@@ -1,35 +1,34 @@
 #/bin/bash
-DESKTOP=/home/liuliang/Desktop
-ROOT=$DESKTOP/pytorch-deep-image-matting-pure
-DATA_ROOT=$DESKTOP/dataset/matting
-TRAIN_DATA_ROOT=$DATA_ROOT/dataset_deep_image_matting/Training_set
-TEST_DATA_ROOT=$DESKTOP/dataset/matting/dataset_deep_image_matting/Test_set
+DATA_ROOT=/data/datasets/matting/Combined_Dataset
+TRAIN_DATA_ROOT=$DATA_ROOT/Training_set/comp
+TEST_DATA_ROOT=$DATA_ROOT/Test_set/comp
 
 python core/train.py \
     --crop_h=320,480,640 \
     --crop_w=320,480,640 \
     --size_h=320 \
     --size_w=320 \
-    --alphaDir=$TRAIN_DATA_ROOT/comp/alpha  \
-    --fgDir=$TRAIN_DATA_ROOT/comp/fg \
-    --bgDir=$TRAIN_DATA_ROOT/comp/bg \
-    --imgDir=$TRAIN_DATA_ROOT/comp/image \
-    --saveDir=$ROOT/model/stage0 \
+    --alphaDir=$TRAIN_DATA_ROOT/alpha  \
+    --fgDir=$TRAIN_DATA_ROOT/fg \
+    --bgDir=$TRAIN_DATA_ROOT/bg \
+    --imgDir=$TRAIN_DATA_ROOT/image \
+    --saveDir=model/stage0 \
     --batchSize=1 \
-    --nEpochs=50 \
+    --nEpochs=25 \
     --step=-1 \
     --lr=0.00001 \
     --wl_weight=0.5 \
     --threads=4 \
-    --printFreq=50 \
+    --printFreq=1 \
     --ckptSaveFreq=1 \
     --cuda \
     --stage=0 \
     --pretrain=model/vgg_state_dict.pth \
     --testFreq=1 \
-    --testImgDir=$TEST_DATA_ROOT/comp/image \
-    --testTrimapDir=$TEST_DATA_ROOT/comp/trimap \
-    --testAlphaDir=$TEST_DATA_ROOT/comp/alpha \
-    --testResDir=$ROOT/result/tmp \
+    --testImgDir=$TEST_DATA_ROOT/image \
+    --testTrimapDir=$TEST_DATA_ROOT/trimap \
+    --testAlphaDir=$TEST_DATA_ROOT/alpha \
+    --testResDir=result/tmp \
     --crop_or_resize=whole \
-    --max_size=320 
+    --max_size=1600
+    #--resume=model/stage0/ckpt_e6.pth \