reformat.js 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. export const fourNodes2xywh = function (data) {
  2. let lines = data.replace('\r', '').split('\n')
  3. let res = []
  4. lines.forEach(line => {
  5. const arr = line.split(' ')
  6. let tmp = []
  7. tmp.push(arr[0])
  8. tmp.push((parseFloat(arr[3]) + parseFloat(arr[5])) / 2)
  9. tmp.push((parseFloat(arr[2]) + parseFloat(arr[4])) / 2)
  10. tmp.push(Math.abs(arr[5] - arr[3]))
  11. tmp.push(Math.abs(arr[4] - arr[2]))
  12. res.push(tmp.join(' '))
  13. })
  14. // console.log(res)
  15. return res.join('\r\n')
  16. }
  17. export const xywh2fourNodes = function (data) {
  18. let lines = data.replace('\r', '').split('\n')
  19. let res = []
  20. lines.forEach(line => {
  21. const arr = line.split(' ')
  22. let tmp = []
  23. tmp.push(arr[0])
  24. tmp.push(parseFloat(arr[1]) + parseFloat(arr[3]) / 2)
  25. tmp.push(parseFloat(arr[2]) - parseFloat(arr[4]) / 2)
  26. tmp.push(parseFloat(arr[1]) + parseFloat(arr[3]) / 2)
  27. tmp.push(parseFloat(arr[2]) + parseFloat(arr[4]) / 2)
  28. tmp.push(parseFloat(arr[1]) - parseFloat(arr[3]) / 2)
  29. tmp.push(parseFloat(arr[2]) + parseFloat(arr[4]) / 2)
  30. tmp.push(parseFloat(arr[1]) - parseFloat(arr[3]) / 2)
  31. tmp.push(parseFloat(arr[2]) - parseFloat(arr[4]) / 2)
  32. res.push(tmp.join(' '))
  33. })
  34. // console.log(res)
  35. return res.join('\r\n')
  36. }