def fromdir(dirPath, gtPath, suffix='.jpg'):
aSequence = Sequence()
aSequence.frames = sorted([framePath for framePath in os.listdir(dirPath) if framePath.endswith(suffix)])
aSequence.path = dirPath
aSequence.marker = 0
aSequence.boxes[gtPath] = benchutils.parse_gt(gtPath)
return aSequence
def fromdir(dirPath, gtPath, suffix='.jpg'):
aSequence = Sequence()
aSequence.frames = sorted([
framePath for framePath in os.listdir(dirPath)
if framePath.endswith(suffix)
])
aSequence.path = dirPath
aSequence.marker = 0
aSequence.boxes[gtPath] = benchutils.parse_gt(gtPath)
return aSequence
def sequential_link(seqDir, textDir, outputDir, excludes=['Football1', 'David', 'Freeman3', 'Freeman4', 'Jogging'], suffix='.jpg', proportion=0.5, pattern='groundtruth_rect.txt'):
sequences = sorted([sequence for sequence in os.listdir(seqDir) if sequence not in excludes])
trainSize = int(len(sequences)*proportion)
randomSequences = list(sequences)
random.shuffle(randomSequences)
trainSequences = randomSequences[:trainSize]
testSequences = randomSequences[trainSize:]
trainDatabase = []
testDatabase = []
databaseGt = []
index = 0
omittedIndexes = []
for sequence in sequences:
if sequence in excludes:
continue
seqPath = os.path.join(seqDir, sequence)
gt = benchutils.parse_gt(os.path.join(seqPath, pattern))
print seqPath
frames = [aFrame for aFrame in sorted(os.listdir(os.path.join(seqPath, 'img'))) if aFrame.endswith(suffix)]
if not len(frames) == len(gt):
raise Exception('Mismatching frame and gt length ({} vs. {})'.format(len(frames), len(gt)))
gtIndex = 0
if sequence in trainSequences:
editDatabase = trainDatabase
elif sequence in testSequences:
editDatabase = testDatabase
for frame in frames:
os.symlink(os.path.join(seqDir, sequence, 'img', frame), os.path.join(outputDir, str(index)+suffix))
if not int(frame.replace(suffix, '')) == 1:
editDatabase.append(index)
else:
print 'Omitted frame {} for sequence {} at index {}'.format(frame, sequence, index)
omittedIndexes.append(index)
databaseGt.append(gt[gtIndex])
gtIndex += 1
index += 1
linkNames = sorted([int(link.replace('.jpg', '')) for link in os.listdir(outputDir)])
print 'Links: {} train: {} test: {} gt: {}'.format(len(linkNames), len(trainDatabase), len(testDatabase), len(databaseGt))
write_database(textDir, 'train.txt', linkNames, trainDatabase)
write_gt(textDir, 'train_gt.txt', linkNames, trainDatabase + omittedIndexes, databaseGt)
write_database(textDir, 'test.txt', linkNames, testDatabase)
write_gt(textDir, 'test_gt.txt', linkNames, testDatabase + omittedIndexes, databaseGt)
gtIndex += 1
index += 1
linkNames = sorted([int(link.replace('.jpg', '')) for link in os.listdir(outputDir)])
print 'Links: {} train: {} test: {} gt: {}'.format(len(linkNames), len(trainDatabase), len(testDatabase), len(databaseGt))
write_database(textDir, 'train.txt', linkNames, trainDatabase)
write_gt(textDir, 'train_gt.txt', linkNames, trainDatabase + omittedIndexes, databaseGt)
write_database(textDir, 'test.txt', linkNames, testDatabase)
write_gt(textDir, 'test_gt.txt', linkNames, testDatabase + omittedIndexes, databaseGt)
if __name__ == '__main__':
if len(sys.argv) < 2:
print 'Usage: python', sys.argv[0], 'sequenceDir'
sys.exit(0)
baseDir = sys.argv[1]
images = sorted([name.strip().replace('.jpg', '') for name in os.listdir(os.path.join(baseDir, 'img')) if name.endswith('.jpg')])
gt = benchutils.parse_gt(os.path.join(baseDir, 'groundtruth_rect.txt'))
size = len(images)
#30 frames ~= 1 seg
trainSamples = 30
if not size == len(gt):
print 'Images: {} GtTs: {}'.format(size, len(gt))
sys.exit(1)
ids = range(size)
random.shuffle(ids)
trainIds = ids[:trainSamples]
testIds = ids[trainSamples:]
all = open(os.path.join(baseDir, 'allImagesList.txt'), 'w')
train = open(os.path.join(baseDir, 'train.txt'), 'w')
test = open(os.path.join(baseDir, 'test.txt'), 'w')
trainGt = open(os.path.join(baseDir, 'train_gt.txt'), 'w')
testGt = open(os.path.join(baseDir, 'test_gt.txt'), 'w')