完成实验四

2024-01-12 02:27:03 +08:00
parent 7fbb893223
commit feca24347a
15 changed files with 4588 additions and 7 deletions
--- a/Lab4/dataset/Vehicles/split_dataset.py
+++ b/Lab4/dataset/Vehicles/split_dataset.py
@@ -0,0 +1,26 @@
+import os
+import random
+import pandas as pd
+
+train_list = list()
+test_list = list()
+
+root_dir = "raw"
+class_index = 0
+for vehicle in os.listdir(root_dir):
+    img_list = [i for i in os.listdir(os.path.join(root_dir, vehicle)) if i.endswith(".jpg")]
+    random.shuffle(img_list)
+    split_num = int(len(img_list) * 0.8)
+    for img in img_list[0 : split_num]:
+        train_list.append([os.path.join(root_dir, vehicle, img), class_index])
+    for img in img_list[split_num : ]:
+        test_list.append([os.path.join(root_dir, vehicle, img), class_index])
+    class_index += 1
+
+train_list.sort()
+test_list.sort()
+
+pd.DataFrame(data=train_list, columns=["Vehicle", "Label"]).to_csv("./train.csv", index=False)
+pd.DataFrame(data=test_list, columns=["Vehicle", "Label"]).to_csv("./test.csv", index=False)
+
+
--- a/Lab4/dataset/Vehicles/test.csv
+++ b/Lab4/dataset/Vehicles/test.csv
@@ -0,0 +1,273 @@
+Vehicle,Label
+raw/bus/bus002.jpg,1
+raw/bus/bus005.jpg,1
+raw/bus/bus008.jpg,1
+raw/bus/bus017.jpg,1
+raw/bus/bus019.jpg,1
+raw/bus/bus033.jpg,1
+raw/bus/bus034.jpg,1
+raw/bus/bus040.jpg,1
+raw/bus/bus043.jpg,1
+raw/bus/bus044.jpg,1
+raw/bus/bus045.jpg,1
+raw/bus/bus046.jpg,1
+raw/bus/bus051.jpg,1
+raw/bus/bus052.jpg,1
+raw/bus/bus056.jpg,1
+raw/bus/bus057.jpg,1
+raw/bus/bus063.jpg,1
+raw/bus/bus066.jpg,1
+raw/bus/bus077.jpg,1
+raw/bus/bus080.jpg,1
+raw/bus/bus096.jpg,1
+raw/bus/bus097.jpg,1
+raw/bus/bus099.jpg,1
+raw/bus/bus114.jpg,1
+raw/bus/bus115.jpg,1
+raw/bus/bus119.jpg,1
+raw/bus/bus127.jpg,1
+raw/bus/bus132.jpg,1
+raw/bus/bus133.jpg,1
+raw/bus/bus135.jpg,1
+raw/bus/bus139.jpg,1
+raw/bus/bus141.jpg,1
+raw/bus/bus145.jpg,1
+raw/bus/bus156.jpg,1
+raw/bus/bus170.jpg,1
+raw/bus/bus173.jpg,1
+raw/bus/bus183.jpg,1
+raw/bus/bus190.jpg,1
+raw/bus/bus192.jpg,1
+raw/bus/bus200.jpg,1
+raw/bus/bus201.jpg,1
+raw/bus/bus209.jpg,1
+raw/bus/bus213.jpg,1
+raw/bus/bus218.jpg,1
+raw/car/car002.jpg,0
+raw/car/car012.jpg,0
+raw/car/car018.jpg,0
+raw/car/car025.jpg,0
+raw/car/car030.jpg,0
+raw/car/car034.jpg,0
+raw/car/car036.jpg,0
+raw/car/car047.jpg,0
+raw/car/car053.jpg,0
+raw/car/car054.jpg,0
+raw/car/car058.jpg,0
+raw/car/car060.jpg,0
+raw/car/car062.jpg,0
+raw/car/car063.jpg,0
+raw/car/car066.jpg,0
+raw/car/car072.jpg,0
+raw/car/car084.jpg,0
+raw/car/car087.jpg,0
+raw/car/car089.jpg,0
+raw/car/car099.jpg,0
+raw/car/car103.jpg,0
+raw/car/car105.jpg,0
+raw/car/car106.jpg,0
+raw/car/car116.jpg,0
+raw/car/car117.jpg,0
+raw/car/car120.jpg,0
+raw/car/car132.jpg,0
+raw/car/car138.jpg,0
+raw/car/car139.jpg,0
+raw/car/car152.jpg,0
+raw/car/car153.jpg,0
+raw/car/car157.jpg,0
+raw/car/car168.jpg,0
+raw/car/car177.jpg,0
+raw/car/car182.jpg,0
+raw/car/car183.jpg,0
+raw/car/car184.jpg,0
+raw/car/car187.jpg,0
+raw/car/car192.jpg,0
+raw/car/car209.jpg,0
+raw/car/car212.jpg,0
+raw/car/car215.jpg,0
+raw/car/car222.jpg,0
+raw/car/car225.jpg,0
+raw/car/car227.jpg,0
+raw/car/car237.jpg,0
+raw/car/car238.jpg,0
+raw/car/car249.jpg,0
+raw/car/car250.jpg,0
+raw/car/car257.jpg,0
+raw/car/car263.jpg,0
+raw/car/car275.jpg,0
+raw/car/car279.jpg,0
+raw/car/car283.jpg,0
+raw/car/car288.jpg,0
+raw/car/car299.jpg,0
+raw/car/car300.jpg,0
+raw/car/car317.jpg,0
+raw/car/car318.jpg,0
+raw/car/car336.jpg,0
+raw/car/car345.jpg,0
+raw/car/car351.jpg,0
+raw/car/car357.jpg,0
+raw/car/car358.jpg,0
+raw/car/car360.jpg,0
+raw/car/car369.jpg,0
+raw/car/car374.jpg,0
+raw/car/car376.jpg,0
+raw/car/car377.jpg,0
+raw/car/car379.jpg,0
+raw/car/car380.jpg,0
+raw/car/car381.jpg,0
+raw/car/car386.jpg,0
+raw/car/car396.jpg,0
+raw/car/car402.jpg,0
+raw/car/car404.jpg,0
+raw/car/car410.jpg,0
+raw/car/car424.jpg,0
+raw/car/car437.jpg,0
+raw/car/car439.jpg,0
+raw/car/car441.jpg,0
+raw/car/car454.jpg,0
+raw/car/car457.jpg,0
+raw/car/car458.jpg,0
+raw/car/car460.jpg,0
+raw/car/car461.jpg,0
+raw/car/car470.jpg,0
+raw/car/car475.jpg,0
+raw/car/car477.jpg,0
+raw/car/car486.jpg,0
+raw/car/car490.jpg,0
+raw/car/car495.jpg,0
+raw/car/car500.jpg,0
+raw/car/car502.jpg,0
+raw/car/car511.jpg,0
+raw/car/car525.jpg,0
+raw/car/car534.jpg,0
+raw/car/car537.jpg,0
+raw/car/car544.jpg,0
+raw/car/car546.jpg,0
+raw/car/car547.jpg,0
+raw/car/car551.jpg,0
+raw/car/car559.jpg,0
+raw/car/car566.jpg,0
+raw/car/car568.jpg,0
+raw/car/car570.jpg,0
+raw/car/car571.jpg,0
+raw/car/car574.jpg,0
+raw/car/car577.jpg,0
+raw/car/car580.jpg,0
+raw/car/car583.jpg,0
+raw/car/car585.jpg,0
+raw/car/car588.jpg,0
+raw/car/car600.jpg,0
+raw/car/car604.jpg,0
+raw/car/car606.jpg,0
+raw/car/car611.jpg,0
+raw/car/car613.jpg,0
+raw/car/car625.jpg,0
+raw/car/car626.jpg,0
+raw/car/car638.jpg,0
+raw/car/car646.jpg,0
+raw/car/car647.jpg,0
+raw/car/car649.jpg,0
+raw/car/car653.jpg,0
+raw/car/car661.jpg,0
+raw/car/car664.jpg,0
+raw/car/car667.jpg,0
+raw/car/car669.jpg,0
+raw/car/car670.jpg,0
+raw/car/car691.jpg,0
+raw/car/car698.jpg,0
+raw/car/car701.jpg,0
+raw/car/car707.jpg,0
+raw/car/car715.jpg,0
+raw/car/car716.jpg,0
+raw/car/car718.jpg,0
+raw/car/car720.jpg,0
+raw/car/car728.jpg,0
+raw/car/car730.jpg,0
+raw/car/car733.jpg,0
+raw/car/car741.jpg,0
+raw/car/car743.jpg,0
+raw/car/car753.jpg,0
+raw/car/car755.jpg,0
+raw/car/car756.jpg,0
+raw/car/car757.jpg,0
+raw/car/car760.jpg,0
+raw/car/car761.jpg,0
+raw/car/car763.jpg,0
+raw/car/car764.jpg,0
+raw/car/car768.jpg,0
+raw/car/car769.jpg,0
+raw/car/car770.jpg,0
+raw/car/car772.jpg,0
+raw/car/car774.jpg,0
+raw/truck/truck001.jpg,2
+raw/truck/truck006.jpg,2
+raw/truck/truck009.jpg,2
+raw/truck/truck011.jpg,2
+raw/truck/truck018.jpg,2
+raw/truck/truck020.jpg,2
+raw/truck/truck021.jpg,2
+raw/truck/truck026.jpg,2
+raw/truck/truck031.jpg,2
+raw/truck/truck039.jpg,2
+raw/truck/truck043.jpg,2
+raw/truck/truck045.jpg,2
+raw/truck/truck051.jpg,2
+raw/truck/truck067.jpg,2
+raw/truck/truck075.jpg,2
+raw/truck/truck080.jpg,2
+raw/truck/truck084.jpg,2
+raw/truck/truck088.jpg,2
+raw/truck/truck091.jpg,2
+raw/truck/truck093.jpg,2
+raw/truck/truck095.jpg,2
+raw/truck/truck097.jpg,2
+raw/truck/truck098.jpg,2
+raw/truck/truck103.jpg,2
+raw/truck/truck107.jpg,2
+raw/truck/truck113.jpg,2
+raw/truck/truck116.jpg,2
+raw/truck/truck117.jpg,2
+raw/truck/truck118.jpg,2
+raw/truck/truck124.jpg,2
+raw/truck/truck130.jpg,2
+raw/truck/truck136.jpg,2
+raw/truck/truck141.jpg,2
+raw/truck/truck146.jpg,2
+raw/truck/truck152.jpg,2
+raw/truck/truck153.jpg,2
+raw/truck/truck155.jpg,2
+raw/truck/truck157.jpg,2
+raw/truck/truck164.jpg,2
+raw/truck/truck170.jpg,2
+raw/truck/truck171.jpg,2
+raw/truck/truck176.jpg,2
+raw/truck/truck185.jpg,2
+raw/truck/truck192.jpg,2
+raw/truck/truck193.jpg,2
+raw/truck/truck197.jpg,2
+raw/truck/truck207.jpg,2
+raw/truck/truck211.jpg,2
+raw/truck/truck216.jpg,2
+raw/truck/truck217.jpg,2
+raw/truck/truck218.jpg,2
+raw/truck/truck220.jpg,2
+raw/truck/truck221.jpg,2
+raw/truck/truck223.jpg,2
+raw/truck/truck226.jpg,2
+raw/truck/truck235.jpg,2
+raw/truck/truck252.jpg,2
+raw/truck/truck259.jpg,2
+raw/truck/truck261.jpg,2
+raw/truck/truck264.jpg,2
+raw/truck/truck271.jpg,2
+raw/truck/truck306.jpg,2
+raw/truck/truck309.jpg,2
+raw/truck/truck311.jpg,2
+raw/truck/truck317.jpg,2
+raw/truck/truck326.jpg,2
+raw/truck/truck328.jpg,2
+raw/truck/truck329.jpg,2
+raw/truck/truck333.jpg,2
+raw/truck/truck335.jpg,2
+raw/truck/truck345.jpg,2
+raw/truck/truck360.jpg,2
--- a/Lab4/dataset/Vehicles/train.csv
+++ b/Lab4/dataset/Vehicles/train.csv