質問編集履歴
3
ソースの追加
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -9,8 +9,8 @@
|
|
|
9
9
|
|
|
10
10
|
def iou(a, b):
|
|
11
11
|
# a, bは矩形を表すリストで、a=[xmin, xmax, ymin, ymax]
|
|
12
|
-
ax_mn, ay_mn, ax_mx, ay_mx = a[
|
|
12
|
+
ax_mn, ay_mn, ax_mx, ay_mx = a[2], a[3], a[4], a[5]
|
|
13
|
-
bx_mn, by_mn, bx_mx, by_mx = b[
|
|
13
|
+
bx_mn, by_mn, bx_mx, by_mx = b[2], b[3], b[4], b[5]
|
|
14
14
|
|
|
15
15
|
a_area = (ax_mx - ax_mn + 1) * (ay_mx - ay_mn + 1)
|
|
16
16
|
b_area = (bx_mx - bx_mn + 1) * (by_mx - by_mn + 1)
|
|
@@ -26,22 +26,26 @@
|
|
|
26
26
|
iou = intersect / (a_area + b_area - intersect)
|
|
27
27
|
return iou
|
|
28
28
|
|
|
29
|
-
def
|
|
29
|
+
def check_ID(IDs, data):
|
|
30
|
-
for k in IDs:#IDsから登録済みIDを一つずつ呼び出し
|
|
30
|
+
for k in IDs:#IDsから登録済みIDを一つずつ呼び出し
|
|
31
|
-
if iou(
|
|
31
|
+
if iou(k, data) <= 0.8:#diffで差分の取得
|
|
32
|
-
return
|
|
32
|
+
return k[0]
|
|
33
|
-
return
|
|
33
|
+
return False
|
|
34
34
|
|
|
35
35
|
|
|
36
|
-
IDs =
|
|
36
|
+
IDs = list()#今までに出てきたIDをkey, それに付随する数値をvalueにして格納する
|
|
37
37
|
with open('out1.csv', 'r', encoding='UTF-8') as f:
|
|
38
38
|
reader = csv.reader(f)
|
|
39
39
|
with open('inp.txt','w', encoding='UTF-8', newline="") as i:
|
|
40
40
|
writer = csv.writer(i)
|
|
41
41
|
for data in reader:
|
|
42
42
|
if "ID:" in data[0]:#一要素目がID番号なら
|
|
43
|
-
|
|
43
|
+
result = check_ID(IDs, data)#重複判定
|
|
44
|
+
if result:
|
|
45
|
+
#ヒットすれば、resultに IDが入っているので、dataを書き換える
|
|
44
|
-
|
|
46
|
+
data[0] = result
|
|
47
|
+
else:
|
|
48
|
+
IDs.append(data) #IDsにまるごと登録 これをelseでなくてに出せば全部登録になります。
|
|
45
49
|
writer.writerow(data)#csv形式で書き込み
|
|
46
50
|
```
|
|
47
51
|
### 試したこと
|
2
ソースの追加
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -9,8 +9,8 @@
|
|
|
9
9
|
|
|
10
10
|
def iou(a, b):
|
|
11
11
|
# a, bは矩形を表すリストで、a=[xmin, xmax, ymin, ymax]
|
|
12
|
-
ax_mn, ay_mn, ax_mx, ay_mx = a[
|
|
12
|
+
ax_mn, ay_mn, ax_mx, ay_mx = a[0], a[1], a[2], a[3]
|
|
13
|
-
bx_mn, by_mn, bx_mx, by_mx = b[
|
|
13
|
+
bx_mn, by_mn, bx_mx, by_mx = b[0], b[1], b[2], b[3]
|
|
14
14
|
|
|
15
15
|
a_area = (ax_mx - ax_mn + 1) * (ay_mx - ay_mn + 1)
|
|
16
16
|
b_area = (bx_mx - bx_mn + 1) * (by_mx - by_mn + 1)
|
|
@@ -28,8 +28,7 @@
|
|
|
28
28
|
|
|
29
29
|
def renumber(IDs, data):
|
|
30
30
|
for k in IDs:#IDsから登録済みIDを一つずつ呼び出し
|
|
31
|
-
diff = [abs(int(v)-int(d)) for v, d in zip(IDs[k], data[2:])]#登録済みのIDと対象のIDの座標の差分をリストに格納(data[2:]はその要素以降の要素)（absで絶対値）
|
|
32
|
-
if
|
|
31
|
+
if iou(a,b) <= 0.8:#diffで差分の取得
|
|
33
32
|
return [k] + data[1:]#ID名を登録済みの番号に変えて値を返す
|
|
34
33
|
return data#登録済みIDと重複がないと判定されたら、何も変更せずに返す
|
|
35
34
|
|
1
ソースの修正
title
CHANGED
|
File without changes
|
body
CHANGED
|
@@ -3,10 +3,29 @@
|
|
|
3
3
|
下に記載するコードを合わせて各行のiou値を比較して一定範囲内なら同じID似直すというプログラムを作りたい
|
|
4
4
|
|
|
5
5
|
### 該当のソースコード
|
|
6
|
-
|
|
7
6
|
```python
|
|
8
7
|
import pandas as pd
|
|
9
8
|
import csv
|
|
9
|
+
|
|
10
|
+
def iou(a, b):
|
|
11
|
+
# a, bは矩形を表すリストで、a=[xmin, xmax, ymin, ymax]
|
|
12
|
+
ax_mn, ay_mn, ax_mx, ay_mx = a[3], a[4], a[5], a[6]
|
|
13
|
+
bx_mn, by_mn, bx_mx, by_mx = b[3], b[4], b[5], b[6]
|
|
14
|
+
|
|
15
|
+
a_area = (ax_mx - ax_mn + 1) * (ay_mx - ay_mn + 1)
|
|
16
|
+
b_area = (bx_mx - bx_mn + 1) * (by_mx - by_mn + 1)
|
|
17
|
+
|
|
18
|
+
abx_mn = max(ax_mn, bx_mn)
|
|
19
|
+
aby_mn = max(ay_mn, by_mn)
|
|
20
|
+
abx_mx = min(ax_mx, bx_mx)
|
|
21
|
+
aby_mx = min(ay_mx, by_mx)
|
|
22
|
+
w = max(0, abx_mx - abx_mn + 1)
|
|
23
|
+
h = max(0, aby_mx - aby_mn + 1)
|
|
24
|
+
intersect = w*h
|
|
25
|
+
|
|
26
|
+
iou = intersect / (a_area + b_area - intersect)
|
|
27
|
+
return iou
|
|
28
|
+
|
|
10
29
|
def renumber(IDs, data):
|
|
11
30
|
for k in IDs:#IDsから登録済みIDを一つずつ呼び出し
|
|
12
31
|
diff = [abs(int(v)-int(d)) for v, d in zip(IDs[k], data[2:])]#登録済みのIDと対象のIDの座標の差分をリストに格納(data[2:]はその要素以降の要素)（absで絶対値）
|
|
@@ -22,31 +41,10 @@
|
|
|
22
41
|
writer = csv.writer(i)
|
|
23
42
|
for data in reader:
|
|
24
43
|
if "ID:" in data[0]:#一要素目がID番号なら
|
|
25
|
-
data = renumber(IDs, data)#
|
|
44
|
+
data = renumber(IDs, data)#重複判定
|
|
26
45
|
IDs[data[0]] = data[2:]#IDsに登録
|
|
27
46
|
writer.writerow(data)#csv形式で書き込み
|
|
28
47
|
```
|
|
29
|
-
```python
|
|
30
|
-
def iou(a, b):
|
|
31
|
-
# a, bは矩形を表すリストで、a=[xmin, xmax, ymin, ymax]
|
|
32
|
-
ax_mn, ay_mn, ax_mx, ay_mx = a[0], a[1], a[2], a[3]
|
|
33
|
-
bx_mn, by_mn, bx_mx, by_mx = b[0], b[1], b[2], b[3]
|
|
34
|
-
|
|
35
|
-
a_area = (ax_mx - ax_mn + 1) * (ay_mx - ay_mn + 1)
|
|
36
|
-
b_area = (bx_mx - bx_mn + 1) * (by_mx - by_mn + 1)
|
|
37
|
-
|
|
38
|
-
abx_mn = max(ax_mn, bx_mn)
|
|
39
|
-
aby_mn = max(ay_mn, by_mn)
|
|
40
|
-
abx_mx = min(ax_mx, bx_mx)
|
|
41
|
-
aby_mx = min(ay_mx, by_mx)
|
|
42
|
-
w = max(0, abx_mx - abx_mn + 1)
|
|
43
|
-
h = max(0, aby_mx - aby_mn + 1)
|
|
44
|
-
intersect = w*h
|
|
45
|
-
|
|
46
|
-
iou = intersect / (a_area + b_area - intersect)
|
|
47
|
-
return iou
|
|
48
|
-
```
|
|
49
|
-
|
|
50
48
|
### 試したこと
|
|
51
49
|
|
|
52
50
|
上のプログラムは各業の座標値を比較して指定した範囲内なら同じID番号にするというプログラムです
|