Mercurial > hg > nsaunier > traffic-intelligence
comparison scripts/classify-objects.py @ 898:1fc901d983ed
better take into account unknown appearance classification
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Wed, 21 Jun 2017 17:21:03 -0400 |
| parents | ff92801e5c54 |
| children | 1466a63dd1cf |
comparison
equal
deleted
inserted
replaced
| 897:f5a49b603e8b | 898:1fc901d983ed |
|---|---|
| 112 if inter.contains(frameNum): | 112 if inter.contains(frameNum): |
| 113 if inter.first == frameNum: | 113 if inter.first == frameNum: |
| 114 obj.initClassifyUserTypeHoGSVM(speedAggregationFunc, pedBikeCarSVM, bikeCarSVM, classifierParams.maxPedestrianSpeed, classifierParams.maxCyclistSpeed, classifierParams.nFramesIgnoreAtEnds) | 114 obj.initClassifyUserTypeHoGSVM(speedAggregationFunc, pedBikeCarSVM, bikeCarSVM, classifierParams.maxPedestrianSpeed, classifierParams.maxCyclistSpeed, classifierParams.nFramesIgnoreAtEnds) |
| 115 currentObjects.append(obj) | 115 currentObjects.append(obj) |
| 116 elif inter.last == frameNum: | 116 elif inter.last == frameNum: |
| 117 obj.classifyUserTypeHoGSVM(minSpeedEquiprobable = classifierParams.minSpeedEquiprobable, speedProbabilities = speedProbabilities) | 117 obj.classifyUserTypeHoGSVM(minSpeedEquiprobable = classifierParams.minSpeedEquiprobable, speedProbabilities = speedProbabilities, maxPercentUnknown = 0.5) # todo add maxPercentUnknown in cfg |
| 118 pastObjects.append(obj) | 118 pastObjects.append(obj) |
| 119 else: | 119 else: |
| 120 obj.classifyUserTypeHoGSVMAtInstant(img, frameNum, invHomography, width, height, classifierParams.percentIncreaseCrop, classifierParams.percentIncreaseCrop, classifierParams.minNPixels, classifierParams.hogRescaleSize, classifierParams.hogNOrientations, classifierParams.hogNPixelsPerCell, classifierParams.hogNCellsPerBlock, classifierParams.hogBlockNorm) | 120 obj.classifyUserTypeHoGSVMAtInstant(img, frameNum, invHomography, width, height, classifierParams.percentIncreaseCrop, classifierParams.percentIncreaseCrop, classifierParams.minNPixels, classifierParams.hogRescaleSize, classifierParams.hogNOrientations, classifierParams.hogNPixelsPerCell, classifierParams.hogNCellsPerBlock, classifierParams.hogBlockNorm) |
| 121 currentObjects.append(obj) | 121 currentObjects.append(obj) |
| 122 else: | 122 else: |
