--- a +++ b/code/exam.py @@ -0,0 +1,25 @@ +""" + Remove some wrong file +""" + +import SimpleITK as sitk +import numpy as np +import csv +from glob import glob +import pandas as pd +import os +import matplotlib +import matplotlib.pyplot as plt +import traceback +import random +from PIL import Image + +path = '/home/ubuntu/data/train-3d/' + +file_list=glob(path+"*.npy") + +for file in file_list: + data = np.load(file) + print(data.shape) + if data.shape != (26,40,40): + os.remove(file) \ No newline at end of file