fixed naming bug
This commit is contained in:
@@ -26,13 +26,13 @@ def h5_sample( lst, sample ):
|
|||||||
# read h5.lst - note - removes // from image column
|
# read h5.lst - note - removes // from image column
|
||||||
print( "reading SwissFEL ls file" )
|
print( "reading SwissFEL ls file" )
|
||||||
# scrub file name
|
# scrub file name
|
||||||
lst_name = os.path.basename( lst )
|
lst_name = os.path.splitext( os.path.basename( lst ) )[0]
|
||||||
|
|
||||||
cols = [ "h5", "image" ]
|
cols = [ "h5", "image" ]
|
||||||
sample_df = pd.read_csv( lst, sep="\s//", engine="python", names=cols )
|
sample_df = pd.read_csv( lst, sep="\s//", engine="python", names=cols )
|
||||||
print( "done" )
|
print( "done" )
|
||||||
|
|
||||||
print( "taking {0} image sample".format( sample ) )
|
print( "taking {0} image sample of {1} image lst".format( sample, len( sample_df ) ) )
|
||||||
# take defined sample
|
# take defined sample
|
||||||
sample_df = sample_df.sample( sample )
|
sample_df = sample_df.sample( sample )
|
||||||
|
|
||||||
@@ -71,3 +71,4 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user