Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13k
stars
4.11k
forks
source link
How to get segmentations in form of [ [670,35][6,305] [60,3]] from segments model of yolov7 #2046
Open
manoj-samal opened 2 weeks ago
How to get segmentations in form of [ [670,35][6,305] [60,3]] from segments model of yolov7 i need in points format
results= model(im, augment=augment, visualize=visualize) i'm getting output in tuple format
results = ( tensor[[[[ 2.55792e-01, -8.02494e-01, -1.52477e-02, ..., 1.31591e-01, 7.27416e-01, -7.37097e-01], [ 4.26252e-01, -5.88908e-01, -2.77992e-01, ..., 5.57948e-02, 4.44748e-01, -7.22842e-01], [-3.02036e-01, -2.36281e-01, -5.49053e-01, ..., -3.83684e-02, 3.37734e-01, -6.16241e-01]]]]], device='cuda:0'),
tensor([[[[[ 0.74621, -0.36439, 0.65680, ..., -0.58979, 0.57147, -0.56773], [-0.17386, -0.39769, 0.95084, ..., -0.61642, 0.48401, -0.48086], [ 0.01340, -0.13087, 1.09881, ..., -0.53496, 0.38888, -0.53385] ]]]], device='cuda:0')]
we can seperate it
pred , out = results