import cv2
import numpy as np
import os
from shutil import copyfile
from sys import exit
train_txt = "../data/ImageSets/train.txt"
trainval_txt = "../data/ImageSets/trainval.txt"
test_txt = "../data/ImageSets/test.txt"
val_txt = "../data/ImageSets/val.txt"

dir_txt = "model/ship_gt/labels/"
# save_dir_ = "model/ship_gt/labels_test/"
save_dir_ = "model/ship_gt/labels_train/"
source_dir = "model/ship_gt/labels_int/"

name_list = [] 

# with open(test_txt, 'r') as fr:
with open(train_txt, 'r') as fr:
    for line in  fr.readlines():
        name_list.append(line.split('\n')[0])
        source = '{}{}.txt'.format(source_dir,line.split('\n')[0])
        target = '{}{}.txt'.format(save_dir_,line.split('\n')[0])
        try:
            copyfile(source, target)
        except IOError as e:
            print("Unable to copy file. %s" % e)
            exit(1)
        except:
            print("Unexpected error:", sys.exc_info())
            exit(1)