Skip to content

Commit d5b1557

Browse files
smithlaismithlai
committed
1.add quick start shell
1 parent ca1429a commit d5b1557

31 files changed

+8885
-0
lines changed

‎MajiangQuickStart.sh‎

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
#copy images and xmls into images folder (into subfolder train and test)
2+
3+
BASE_DIR="$(cd "$(dirname "$0")"; pwd)";
4+
echo "BASE_DIR => $BASE_DIR";
5+
MODELS_DIR="${BASE_DIR}/../models";
6+
echo "MODELS_DIR => $MODELS_DIR";
7+
export PYTHONPATH="${MODELS_DIR}:${MODELS_DIR}/research/:${MODELS_DIR}/research/slim:$PYTHONPATH"
8+
9+
if [ -z "$1" ]
10+
then
11+
echo "No argument supplied: Dataset folder name"
12+
exit 0
13+
fi
14+
#This will read all xmls into one csv file(train_labels and test_labels).
15+
python3 xml_to_csv.py -s "$1"
16+
17+
#python3 pbtxt.py -s "$1"
18+
#Combine the csv and images into one tfrecord file
19+
python3 generate_tfrecord.py --csv_input="$1"/train_labels.csv --image_dir="$1"/train --output_path="$1"/train.record
20+
python3 generate_tfrecord.py --csv_input="$1"/test_labels.csv --image_dir="$1"/test --output_path="$1"/test.record
21+
22+
#train data according to config file data/pipeline_v2.config
23+
python3 ${MODELS_DIR}/research/object_detection/legacy/train.py --logtostderr --train_dir="$1"/training/ --pipeline_config_path="$1"/pipeline_v2.config
24+
25+

‎majiang_data/predefined_classes.txt‎

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
1W
2+
2W
3+
3W
4+
4W
5+
5W
6+
6W
7+
7W
8+
8W
9+
9W
10+
1o
11+
2o
12+
3o
13+
4o
14+
5o
15+
6o
16+
7o
17+
8o
18+
9o
19+
1i
20+
2i
21+
3i
22+
4i
23+
5i
24+
6i
25+
7i
26+
8i
27+
9i
28+
East
29+
West
30+
North
31+
South
32+
Red
33+
Green
34+
White
35+
36+

‎majiang_data/test/1850361716_882923537.jpg‎

74 KB
Loading

‎majiang_data/test/1850361716_882923537.xml‎

Lines changed: 326 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,326 @@
1+
<annotation>
2+
<folder>data</folder>
3+
<filename>1850361716_882923537.jpg</filename>
4+
<path>C:\Users\Smith\Downloads\windows_v1.8.0\data\1850361716_882923537.jpg</path>
5+
<source>
6+
<database>Unknown</database>
7+
</source>
8+
<size>
9+
<width>756</width>
10+
<height>484</height>
11+
<depth>3</depth>
12+
</size>
13+
<segmented>0</segmented>
14+
<object>
15+
<name>1W</name>
16+
<pose>Unspecified</pose>
17+
<truncated>0</truncated>
18+
<difficult>0</difficult>
19+
<bndbox>
20+
<xmin>13</xmin>
21+
<ymin>54</ymin>
22+
<xmax>67</xmax>
23+
<ymax>123</ymax>
24+
</bndbox>
25+
</object>
26+
<object>
27+
<name>2W</name>
28+
<pose>Unspecified</pose>
29+
<truncated>0</truncated>
30+
<difficult>0</difficult>
31+
<bndbox>
32+
<xmin>75</xmin>
33+
<ymin>56</ymin>
34+
<xmax>113</xmax>
35+
<ymax>120</ymax>
36+
</bndbox>
37+
</object>
38+
<object>
39+
<name>3W</name>
40+
<pose>Unspecified</pose>
41+
<truncated>0</truncated>
42+
<difficult>0</difficult>
43+
<bndbox>
44+
<xmin>133</xmin>
45+
<ymin>56</ymin>
46+
<xmax>177</xmax>
47+
<ymax>118</ymax>
48+
</bndbox>
49+
</object>
50+
<object>
51+
<name>4W</name>
52+
<pose>Unspecified</pose>
53+
<truncated>0</truncated>
54+
<difficult>0</difficult>
55+
<bndbox>
56+
<xmin>190</xmin>
57+
<ymin>59</ymin>
58+
<xmax>229</xmax>
59+
<ymax>116</ymax>
60+
</bndbox>
61+
</object>
62+
<object>
63+
<name>5W</name>
64+
<pose>Unspecified</pose>
65+
<truncated>0</truncated>
66+
<difficult>0</difficult>
67+
<bndbox>
68+
<xmin>246</xmin>
69+
<ymin>59</ymin>
70+
<xmax>286</xmax>
71+
<ymax>118</ymax>
72+
</bndbox>
73+
</object>
74+
<object>
75+
<name>6W</name>
76+
<pose>Unspecified</pose>
77+
<truncated>0</truncated>
78+
<difficult>0</difficult>
79+
<bndbox>
80+
<xmin>299</xmin>
81+
<ymin>54</ymin>
82+
<xmax>339</xmax>
83+
<ymax>116</ymax>
84+
</bndbox>
85+
</object>
86+
<object>
87+
<name>7W</name>
88+
<pose>Unspecified</pose>
89+
<truncated>0</truncated>
90+
<difficult>0</difficult>
91+
<bndbox>
92+
<xmin>355</xmin>
93+
<ymin>54</ymin>
94+
<xmax>393</xmax>
95+
<ymax>115</ymax>
96+
</bndbox>
97+
</object>
98+
<object>
99+
<name>8W</name>
100+
<pose>Unspecified</pose>
101+
<truncated>0</truncated>
102+
<difficult>0</difficult>
103+
<bndbox>
104+
<xmin>412</xmin>
105+
<ymin>56</ymin>
106+
<xmax>452</xmax>
107+
<ymax>116</ymax>
108+
</bndbox>
109+
</object>
110+
<object>
111+
<name>9W</name>
112+
<pose>Unspecified</pose>
113+
<truncated>0</truncated>
114+
<difficult>0</difficult>
115+
<bndbox>
116+
<xmin>467</xmin>
117+
<ymin>54</ymin>
118+
<xmax>508</xmax>
119+
<ymax>117</ymax>
120+
</bndbox>
121+
</object>
122+
<object>
123+
<name>1i</name>
124+
<pose>Unspecified</pose>
125+
<truncated>0</truncated>
126+
<difficult>0</difficult>
127+
<bndbox>
128+
<xmin>70</xmin>
129+
<ymin>156</ymin>
130+
<xmax>113</xmax>
131+
<ymax>219</ymax>
132+
</bndbox>
133+
</object>
134+
<object>
135+
<name>2i</name>
136+
<pose>Unspecified</pose>
137+
<truncated>0</truncated>
138+
<difficult>0</difficult>
139+
<bndbox>
140+
<xmin>128</xmin>
141+
<ymin>157</ymin>
142+
<xmax>169</xmax>
143+
<ymax>223</ymax>
144+
</bndbox>
145+
</object>
146+
<object>
147+
<name>3i</name>
148+
<pose>Unspecified</pose>
149+
<truncated>0</truncated>
150+
<difficult>0</difficult>
151+
<bndbox>
152+
<xmin>184</xmin>
153+
<ymin>157</ymin>
154+
<xmax>226</xmax>
155+
<ymax>220</ymax>
156+
</bndbox>
157+
</object>
158+
<object>
159+
<name>4i</name>
160+
<pose>Unspecified</pose>
161+
<truncated>0</truncated>
162+
<difficult>0</difficult>
163+
<bndbox>
164+
<xmin>240</xmin>
165+
<ymin>158</ymin>
166+
<xmax>280</xmax>
167+
<ymax>222</ymax>
168+
</bndbox>
169+
</object>
170+
<object>
171+
<name>5i</name>
172+
<pose>Unspecified</pose>
173+
<truncated>0</truncated>
174+
<difficult>0</difficult>
175+
<bndbox>
176+
<xmin>292</xmin>
177+
<ymin>157</ymin>
178+
<xmax>334</xmax>
179+
<ymax>221</ymax>
180+
</bndbox>
181+
</object>
182+
<object>
183+
<name>6i</name>
184+
<pose>Unspecified</pose>
185+
<truncated>0</truncated>
186+
<difficult>0</difficult>
187+
<bndbox>
188+
<xmin>349</xmin>
189+
<ymin>159</ymin>
190+
<xmax>392</xmax>
191+
<ymax>225</ymax>
192+
</bndbox>
193+
</object>
194+
<object>
195+
<name>7i</name>
196+
<pose>Unspecified</pose>
197+
<truncated>0</truncated>
198+
<difficult>0</difficult>
199+
<bndbox>
200+
<xmin>405</xmin>
201+
<ymin>158</ymin>
202+
<xmax>449</xmax>
203+
<ymax>219</ymax>
204+
</bndbox>
205+
</object>
206+
<object>
207+
<name>8i</name>
208+
<pose>Unspecified</pose>
209+
<truncated>0</truncated>
210+
<difficult>0</difficult>
211+
<bndbox>
212+
<xmin>461</xmin>
213+
<ymin>157</ymin>
214+
<xmax>505</xmax>
215+
<ymax>222</ymax>
216+
</bndbox>
217+
</object>
218+
<object>
219+
<name>9i</name>
220+
<pose>Unspecified</pose>
221+
<truncated>0</truncated>
222+
<difficult>0</difficult>
223+
<bndbox>
224+
<xmin>519</xmin>
225+
<ymin>157</ymin>
226+
<xmax>560</xmax>
227+
<ymax>219</ymax>
228+
</bndbox>
229+
</object>
230+
<object>
231+
<name>8W</name>
232+
<pose>Unspecified</pose>
233+
<truncated>0</truncated>
234+
<difficult>0</difficult>
235+
<bndbox>
236+
<xmin>648</xmin>
237+
<ymin>61</ymin>
238+
<xmax>714</xmax>
239+
<ymax>149</ymax>
240+
</bndbox>
241+
</object>
242+
<object>
243+
<name>East</name>
244+
<pose>Unspecified</pose>
245+
<truncated>0</truncated>
246+
<difficult>0</difficult>
247+
<bndbox>
248+
<xmin>172</xmin>
249+
<ymin>370</ymin>
250+
<xmax>224</xmax>
251+
<ymax>440</ymax>
252+
</bndbox>
253+
</object>
254+
<object>
255+
<name>South</name>
256+
<pose>Unspecified</pose>
257+
<truncated>0</truncated>
258+
<difficult>0</difficult>
259+
<bndbox>
260+
<xmin>234</xmin>
261+
<ymin>377</ymin>
262+
<xmax>279</xmax>
263+
<ymax>434</ymax>
264+
</bndbox>
265+
</object>
266+
<object>
267+
<name>West</name>
268+
<pose>Unspecified</pose>
269+
<truncated>0</truncated>
270+
<difficult>0</difficult>
271+
<bndbox>
272+
<xmin>293</xmin>
273+
<ymin>383</ymin>
274+
<xmax>336</xmax>
275+
<ymax>437</ymax>
276+
</bndbox>
277+
</object>
278+
<object>
279+
<name>North</name>
280+
<pose>Unspecified</pose>
281+
<truncated>0</truncated>
282+
<difficult>0</difficult>
283+
<bndbox>
284+
<xmin>354</xmin>
285+
<ymin>383</ymin>
286+
<xmax>399</xmax>
287+
<ymax>433</ymax>
288+
</bndbox>
289+
</object>
290+
<object>
291+
<name>Red</name>
292+
<pose>Unspecified</pose>
293+
<truncated>0</truncated>
294+
<difficult>0</difficult>
295+
<bndbox>
296+
<xmin>415</xmin>
297+
<ymin>377</ymin>
298+
<xmax>456</xmax>
299+
<ymax>441</ymax>
300+
</bndbox>
301+
</object>
302+
<object>
303+
<name>Green</name>
304+
<pose>Unspecified</pose>
305+
<truncated>0</truncated>
306+
<difficult>0</difficult>
307+
<bndbox>
308+
<xmin>469</xmin>
309+
<ymin>382</ymin>
310+
<xmax>519</xmax>
311+
<ymax>436</ymax>
312+
</bndbox>
313+
</object>
314+
<object>
315+
<name>White</name>
316+
<pose>Unspecified</pose>
317+
<truncated>0</truncated>
318+
<difficult>0</difficult>
319+
<bndbox>
320+
<xmin>523</xmin>
321+
<ymin>373</ymin>
322+
<xmax>571</xmax>
323+
<ymax>444</ymax>
324+
</bndbox>
325+
</object>
326+
</annotation>

‎majiang_data/test/1850361723_882923537.jpg‎

53.1 KB
Loading

0 commit comments

Comments
 (0)