AlexeyAB
2018-06-04 1e512b23089b23e91fa5ed6a617641d9286d1417
Letter box is disabled in the darknet.py by defualt
4 files modified
9 ■■■■■ changed files
build/darknet/x64/darknet.py 2 ●●● patch | view | raw | blame | history
darknet.py 2 ●●● patch | view | raw | blame | history
src/image.h 2 ●●● patch | view | raw | blame | history
src/network.c 3 ●●●● patch | view | raw | blame | history
build/darknet/x64/darknet.py
@@ -235,7 +235,7 @@
    if debug: print("Assigned pnum")
    predict_image(net, im)
    if debug: print("did prediction")
    dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, None, 0, pnum, 1)
    dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, None, 0, pnum, 0)
    if debug: print("Got dets")
    num = pnum[0]
    if debug: print("got zeroth index of pnum")
darknet.py
@@ -235,7 +235,7 @@
    if debug: print("Assigned pnum")
    predict_image(net, im)
    if debug: print("did prediction")
    dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, None, 0, pnum, 1)
    dets = get_network_boxes(net, im.w, im.h, thresh, hier_thresh, None, 0, pnum, 0)
    if debug: print("Got dets")
    num = pnum[0]
    if debug: print("got zeroth index of pnum")
src/image.h
@@ -30,7 +30,7 @@
image random_crop_image(image im, int w, int h);
image random_augment_image(image im, float angle, float aspect, int low, int high, int size);
void random_distort_image(image im, float hue, float saturation, float exposure);
image resize_image(image im, int w, int h);
YOLODLL_API image resize_image(image im, int w, int h);
void fill_image(image m, float s);
void letterbox_image_into(image im, int w, int h, image boxed);
YOLODLL_API image letterbox_image(image im, int w, int h);
src/network.c
@@ -640,7 +640,8 @@
float *network_predict_image(network *net, image im)
{
    image imr = letterbox_image(im, net->w, net->h);
    //image imr = letterbox_image(im, net->w, net->h);
    image imr = resize_image(im, net->w, net->h);
    set_batch_network(net, 1);
    float *p = network_predict(*net, imr.data);
    free_image(imr);