def main():
sheet_path = './resources/sheets/training/'
output_path = './resources/sheets/training/trimmed/'
reject_parth = './resources/sheets/training/rejected/'
for filename in os.listdir(sheet_path):
if filename.endswith('.jpg') or filename.endswith(
'.jepg') or filename.endswith('.png'):
print(filename)
segmenter = Segmenter(os.path.join(sheet_path, filename))
staff_removed = segmenter.remove_staff_lines()
boxed = segmenter.getSymbols(merge_overlap=True)
cv2.imwrite(output_path + 'boxed{}.png'.format(filename[:-6]),
boxed)
segmenter.saveSymbols(save_origonal=False,
format=filename[:-6] + '_note{:03}.png',
width=150,
path=output_path,
reject_path=reject_parth,
dirty_times=0)
def label_symbols(self,
min_x_pos=500,
edit_last=False,
dont_use_cache=False):
measures = self.xmldoc.getElementsByTagName('Measure')
xml_objects = []
while len(measures) is not 0:
current_measure = measures.pop(0)
current_measure_number = current_measure.getAttribute('number')
for child in current_measure.childNodes:
if child.localName is not None:
if child.nodeName == 'Clef':
xml_objects.append(('clef', current_measure_number))
elif child.nodeName == 'TimeSig':
xml_objects.append(('timeSig', current_measure_number))
elif child.nodeName == 'Rest':
xml_objects.append(('rest_{}'.format(
child.childNodes[1].childNodes[0].data),
current_measure_number))
elif child.nodeName == 'Chord':
if child.childNodes[1].nodeName == 'BeamMode':
xml_objects.append(('tied_{}'.format(
child.childNodes[3].childNodes[0].data),
current_measure_number))
elif child.childNodes[3].nodeName != 'Beam':
xml_objects.append(('{}'.format(
child.childNodes[1].childNodes[0].data),
current_measure_number))
xml_objects.append(('barline', current_measure_number))
first_pic = True
for filename in self.picture_filenames:
segmenter = Segmenter.load_segmenter_from_file(filename)
if (segmenter is None) or dont_use_cache:
segmenter = Segmenter(filename)
if not first_pic:
if edit_last or filename != self.picture_filenames[-1]:
segmenter.blockout_markings()
segmenter.remove_staff_lines()
segmenter.getSymbols(True)
segmenter.save_to_file()
symbols = segmenter.symbols
i = -1
saveCount = 0
while len(symbols) != 0:
i += 1
saveCount = saveCount - 1 if saveCount > 0 else 0
if not is_barline(
symbols[0]) and xml_objects[0][0] == 'barline':
saveCount = 0
print(
'error not enough xmls found at bar {} line {} in file {}'
.format(self.matched_pairs[-1][0][1],
symbols[0].line_num, filename[-20:-4]))
popped = self.matched_pairs.pop()
while popped[0][0] != 'barline':
popped = self.matched_pairs.pop()
self.matched_pairs.append(popped)
popped = symbols.pop(0)
while not is_barline(popped):
popped = symbols.pop(0)
symbols.insert(0, popped)
if is_barline(symbols[0]) and xml_objects[0][0] != 'barline':
saveCount = 0
print(
'error too many xmls found at bar {} line {} in file {}'
.format(self.matched_pairs[-1][0][1],
symbols[0].line_num, filename[-20:-4]))
# save = True
popped = self.matched_pairs.pop()
while popped[0][0] != 'barline':
popped = self.matched_pairs.pop()
self.matched_pairs.append(popped)
popped = xml_objects.pop(0)
while popped[0] != 'barline':
popped = xml_objects.pop(0)
xml_objects.insert(0, popped)
if symbols[0].x < min_x_pos and (not first_pic or i != 0):
self.matched_pairs.append((('clef', -1), symbols.pop(0)))
else:
self.matched_pairs.append(
(xml_objects.pop(0), symbols.pop(0)))
if saveCount > 0:
thumbnail_name = os.path.join(
self.output_path, '{}_{}_line{}__{}.png'.format(
filename[-13:-10], self.matched_pairs[-1][0][0],
self.matched_pairs[-1][1].line_num, i))
cv2.imwrite(thumbnail_name, self.matched_pairs[-1][1].im)
first_pic = False