apps/go-darknet
1
0
Fork 0
mirror of https://github.com/LdDl/go-darknet.git synced 2025-10-05 15:46:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

need to deal with image

Browse Source
This commit is contained in:
Dimitrii
2020-02-19 14:52:19 +03:00
parent 30fbbb9949
commit 2b2f3f159c
20 changed files with 4015 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
example/main.go
View File

@@ -2,6 +2,7 @@ package main
import (
"flag"
"fmt"
"image/jpeg"
"log"
"os"
@@ -37,7 +38,7 @@ func main() {
DataConfigurationFile: *dataConfigFile,
NetworkConfigurationFile: *configFile,
WeightsFile: *weightsFile,
Threshold: .5,
Threshold: .25,
}
if err := n.Init(); err != nil {
printError(err)
@@ -54,6 +55,7 @@ func main() {
if err != nil {
panic(err.Error())
}
_ = src
imgDarknet, err := darknet.Image2Float32(src)
if err != nil {
@@ -67,18 +69,18 @@ func main() {
return
}
_ = dr
// log.Println("Network-only time taken:", dr.NetworkOnlyTimeTaken)
// log.Println("Overall time taken:", dr.OverallTimeTaken, len(dr.Detections))
// for _, d := range dr.Detections {
// for i := range d.ClassIDs {
// bBox := d.BoundingBox
// fmt.Printf("%s (%d): %.4f%% | start point: (%d,%d) | end point: (%d, %d)\n",
// d.ClassNames[i], d.ClassIDs[i],
// d.Probabilities[i],
// bBox.StartPoint.X, bBox.StartPoint.Y,
// bBox.EndPoint.X, bBox.EndPoint.Y,
// )
// }
// }
log.Println("Network-only time taken:", dr.NetworkOnlyTimeTaken)
log.Println("Overall time taken:", dr.OverallTimeTaken, len(dr.Detections))
for _, d := range dr.Detections {
for i := range d.ClassIDs {
bBox := d.BoundingBox
fmt.Printf("%s (%d): %.4f%% | start point: (%d,%d) | end point: (%d, %d)\n",
d.ClassNames[i], d.ClassIDs[i],
d.Probabilities[i],
bBox.StartPoint.X, bBox.StartPoint.Y,
bBox.EndPoint.X, bBox.EndPoint.Y,
)
}
}
}