-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcopy_from_csv.py
107 lines (78 loc) · 3.56 KB
/
copy_from_csv.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# -*- coding: utf-8 -*-
"""
Created on Tue Jul 5 10:59:21 2022
@author: binda
"""
import os
import shutil
import csv
def train_image(csvfile, destination_folder):
#csvfile = 'D:/RID-master/RID-master/venv/segmentation_model_data/train_filenames_1_6_classes.csv'
#source_folder = 'D:/RID-master/RID-master/venv/data/augment_image_outputjhhj/'
#destination_folder = 'D:/RID-master/RID-master/venv/data/csv_datasets/x_train_folder/'
if not os.path.isdir(destination_folder):
os.mkdir(destination_folder)
with open(csvfile,'r') as csvfile:
reader = csv.reader(csvfile, delimiter=",")
#header = next(reader)
#print(header[0])
for row in (reader):
source = row[2]
destination = destination_folder + os.path.basename(source)
#print(destination)
if os.path.isfile(source):
shutil.copy(source, destination)
#print('Moved: ', row)
def test_image(csvfile, destination_folder):
#csvfile = 'D:/RID-master/RID-master/venv/segmentation_model_data/train_filenames_1_6_classes.csv'
#source_folder = 'D:/RID-master/RID-master/venv/data/augment_image_outputjhhj/'
#destination_folder = 'D:/RID-master/RID-master/venv/data/csv_datasets/x_train_folder/'
if not os.path.isdir(destination_folder):
os.mkdir(destination_folder)
with open(csvfile,'r') as csvfile:
reader = csv.reader(csvfile, delimiter=",")
#header = next(reader)
#print(header[0])
for row in (reader):
source = row[2]
destination = destination_folder + os.path.basename(source)
#print(destination)
if os.path.isfile(source):
shutil.copy(source, destination)
#print('Moved: ', row)
def train_mask(csvfile, destination_folder):
#csvfile = 'D:/RID-master/RID-master/venv/segmentation_model_data/train_filenames_1_6_classes.csv'
#source_folder = 'D:/RID-master/RID-master/venv/data/augment_image_outputjhhj/'
#destination_folder = 'D:/RID-master/RID-master/venv/data/csv_datasets/x_train_folder/'
if not os.path.isdir(destination_folder):
os.mkdir(destination_folder)
with open(csvfile,'r') as csvfile:
reader = csv.reader(csvfile, delimiter=",")
#header = next(reader)
#print(header[0])
for row in (reader):
source = row[1]
#print(source)
destination = destination_folder + os.path.basename(source)
#print(destination)
if os.path.isfile(source):
shutil.copy(source, destination)
#print('Moved: ', row)
def test_mask(csvfile, destination_folder):
#csvfile = 'D:/RID-master/RID-master/venv/segmentation_model_data/train_filenames_1_6_classes.csv'
#source_folder = 'D:/RID-master/RID-master/venv/data/augment_image_outputjhhj/'
#destination_folder = 'D:/RID-master/RID-master/venv/data/csv_datasets/x_train_folder/'
if not os.path.isdir(destination_folder):
os.mkdir(destination_folder)
with open(csvfile,'r') as csvfile:
reader = csv.reader(csvfile, delimiter=",")
#header = next(reader)
#print(header[0])
for row in (reader):
source = row[1]
#print(source)
destination = destination_folder + os.path.basename(source)
#print(destination)
if os.path.isfile(source):
shutil.copy(source, destination)
#print('Moved: ', row)