import sys
sys.path.insert(0, '../')
import shutil
import os
import csv
from util import Util
included_cols = [8]
images_source = '/media/willgluck/a2aa6a5f-a88a-45c7-af45-d38ccf2b7639/work/DATASET/test/images/raw/'
images = Util.sort_nicely(os.listdir(images_source))
with open('/media/willgluck/a2aa6a5f-a88a-45c7-af45-d38ccf2b7639/work/SDDS/zoo2MainSpecz.csv', newline='') as csvfile:
spamreader = csv.reader(csvfile, delimiter=',')
for row in spamreader:
if spamreader.line_num >= 1006 and spamreader.line_num <= 2006:
content = list(row[i] for i in included_cols)
if content:
if str(spamreader.line_num - 1) + '.jpg' in images:
if not (content[0].startswith('E') or content[0].startswith('S')):
shutil.move(
'/media/willgluck/a2aa6a5f-a88a-45c7-af45-d38ccf2b7639/work/DATASET/test/images/raw/' + str(spamreader.line_num) + '.jpg',
'/media/willgluck/a2aa6a5f-a88a-45c7-af45-d38ccf2b7639/work/DATASET/test/images/trash/'
)
def load_images_names(self, path, image_dimension):
#'/media/willgluck/a2aa6a5f-a88a-45c7-af45-d38ccf2b7639/work/Galaxies/images/'
self.images_path = path
self.image_dimension = image_dimension * image_dimension
self.images_names = Util.sort_nicely(os.listdir(path))
print("images count: " + str(len(self.images_names)))
