完成实验四

2024-01-12 02:27:03 +08:00
parent 7fbb893223
commit feca24347a
15 changed files with 4588 additions and 7 deletions
--- a/Lab4/dataset/Haze/split.csv
+++ b/Lab4/dataset/Haze/split.csv
@@ -0,0 +1,521 @@
+Image,Train
+001.jpg,1
+002.jpg,1
+003.jpg,1
+004.jpg,0
+005.jpg,0
+006.jpg,1
+007.jpg,1
+008.jpg,1
+009.jpg,1
+010.jpg,1
+011.jpg,0
+012.jpg,1
+013.jpg,0
+014.jpg,1
+015.jpg,1
+016.jpg,1
+017.jpg,1
+018.jpg,1
+019.jpg,1
+020.jpg,1
+021.jpg,1
+022.jpg,1
+023.jpg,1
+024.jpg,1
+025.jpg,1
+026.jpg,1
+027.jpg,0
+028.jpg,1
+029.jpg,1
+030.jpg,1
+031.jpg,1
+032.jpg,1
+033.jpg,1
+034.jpg,1
+035.jpg,1
+036.jpg,1
+037.jpg,0
+038.jpg,1
+039.jpg,1
+040.jpg,1
+041.jpg,1
+042.jpg,1
+043.jpg,0
+044.jpg,1
+045.jpg,1
+046.jpg,1
+047.jpg,1
+048.jpg,0
+049.jpg,1
+050.jpg,1
+051.jpg,1
+052.jpg,1
+053.jpg,1
+054.jpg,1
+055.jpg,1
+056.jpg,1
+057.jpg,1
+058.jpg,1
+059.jpg,0
+060.jpg,1
+061.jpg,1
+062.jpg,1
+063.jpg,1
+064.jpg,1
+065.jpg,0
+066.jpg,1
+067.jpg,0
+068.jpg,1
+069.jpg,0
+070.jpg,1
+071.jpg,1
+072.jpg,1
+073.jpg,1
+074.jpg,1
+075.jpg,1
+076.jpg,0
+077.jpg,1
+078.jpg,1
+079.jpg,1
+080.jpg,1
+081.jpg,1
+082.jpg,1
+083.jpg,1
+084.jpg,1
+085.jpg,0
+086.jpg,1
+087.jpg,1
+088.jpg,0
+089.jpg,1
+090.jpg,1
+091.jpg,1
+092.jpg,1
+093.jpg,1
+094.jpg,1
+095.jpg,1
+096.jpg,1
+097.jpg,1
+098.jpg,1
+099.jpg,1
+100.jpg,1
+101.jpg,1
+102.jpg,1
+103.jpg,1
+104.jpg,1
+105.jpg,1
+106.jpg,1
+107.jpg,0
+108.jpg,1
+109.jpg,0
+110.jpg,1
+111.jpg,1
+112.jpg,1
+113.jpg,1
+114.jpg,1
+115.jpg,1
+116.jpg,1
+117.jpg,1
+118.jpg,0
+119.jpg,1
+120.jpg,1
+121.jpg,0
+122.jpg,1
+123.jpg,1
+124.jpg,1
+125.jpg,1
+126.jpg,0
+127.jpg,1
+128.jpg,0
+129.jpg,0
+130.jpg,0
+131.jpg,1
+132.jpg,1
+133.jpg,1
+134.jpg,1
+135.jpg,1
+136.jpg,1
+137.jpg,1
+138.jpg,1
+139.jpg,1
+140.jpg,0
+141.jpg,0
+142.jpg,1
+143.jpg,1
+144.jpg,1
+145.jpg,1
+146.jpg,1
+147.jpg,0
+148.jpg,1
+149.jpg,1
+150.jpg,0
+151.jpg,1
+152.jpg,1
+153.jpg,1
+154.jpg,1
+155.jpg,0
+156.jpg,1
+157.jpg,0
+158.jpg,1
+159.jpg,0
+160.jpg,1
+161.jpg,0
+162.jpg,1
+163.jpg,0
+164.jpg,1
+165.jpg,0
+166.jpg,1
+167.jpg,1
+168.jpg,0
+169.jpg,1
+170.jpg,1
+171.jpg,0
+172.jpg,1
+173.jpg,1
+174.jpg,1
+175.jpg,0
+176.jpg,1
+177.jpg,1
+178.jpg,1
+179.jpg,1
+180.jpg,0
+181.jpg,1
+182.jpg,0
+183.jpg,1
+184.jpg,1
+185.jpg,1
+186.jpg,0
+187.jpg,1
+188.jpg,1
+189.jpg,1
+190.jpg,0
+191.jpg,1
+192.jpg,1
+193.jpg,1
+194.jpg,1
+195.jpg,1
+196.jpg,1
+197.jpg,1
+198.jpg,1
+199.jpg,1
+200.jpg,1
+201.jpg,1
+202.jpg,1
+203.jpg,1
+204.jpg,1
+205.jpg,0
+206.jpg,1
+207.jpg,1
+208.jpg,1
+209.jpg,1
+210.jpg,1
+211.jpg,1
+212.jpg,1
+213.jpg,1
+214.jpg,0
+215.jpg,0
+216.jpg,1
+217.jpg,1
+218.jpg,1
+219.jpg,1
+220.jpg,1
+221.jpg,1
+222.jpg,1
+223.jpg,0
+224.jpg,0
+225.jpg,1
+226.jpg,1
+227.jpg,1
+228.jpg,1
+229.jpg,0
+230.jpg,1
+231.jpg,1
+232.jpg,1
+233.jpg,0
+234.jpg,1
+235.jpg,0
+236.jpg,1
+237.jpg,1
+238.jpg,1
+239.jpg,1
+240.jpg,1
+241.jpg,1
+242.jpg,1
+243.jpg,1
+244.jpg,1
+245.jpg,1
+246.jpg,0
+247.jpg,1
+248.jpg,0
+249.jpg,0
+250.jpg,1
+251.jpg,1
+252.jpg,1
+253.jpg,1
+254.jpg,1
+255.jpg,1
+256.jpg,0
+257.jpg,1
+258.jpg,1
+259.jpg,1
+260.jpg,0
+261.jpg,1
+262.jpg,0
+263.jpg,1
+264.jpg,1
+265.jpg,0
+266.jpg,0
+267.jpg,0
+268.jpg,1
+269.jpg,1
+270.jpg,1
+271.jpg,1
+272.jpg,1
+273.jpg,1
+274.jpg,1
+275.jpg,1
+276.jpg,1
+277.jpg,1
+278.jpg,1
+279.jpg,1
+280.jpg,0
+281.jpg,1
+282.jpg,1
+283.jpg,1
+284.jpg,1
+285.jpg,1
+286.jpg,0
+287.jpg,1
+288.jpg,1
+289.jpg,1
+290.jpg,1
+291.jpg,1
+292.jpg,1
+293.jpg,0
+294.jpg,1
+295.jpg,1
+296.jpg,0
+297.jpg,1
+298.jpg,1
+299.jpg,0
+300.jpg,1
+301.jpg,1
+302.jpg,0
+303.jpg,1
+304.jpg,0
+305.jpg,1
+306.jpg,1
+307.jpg,1
+308.jpg,1
+309.jpg,1
+310.jpg,1
+311.jpg,0
+312.jpg,1
+313.jpg,1
+314.jpg,0
+315.jpg,1
+316.jpg,1
+317.jpg,1
+318.jpg,1
+319.jpg,1
+320.jpg,1
+321.jpg,0
+322.jpg,1
+323.jpg,1
+324.jpg,0
+325.jpg,1
+326.jpg,1
+327.jpg,0
+328.jpg,1
+329.jpg,1
+330.jpg,1
+331.jpg,1
+332.jpg,0
+333.jpg,1
+334.jpg,0
+335.jpg,1
+336.jpg,1
+337.jpg,1
+338.jpg,1
+339.jpg,1
+340.jpg,1
+341.jpg,1
+342.jpg,1
+343.jpg,1
+344.jpg,1
+345.jpg,0
+346.jpg,1
+347.jpg,1
+348.jpg,1
+349.jpg,1
+350.jpg,1
+351.jpg,1
+352.jpg,1
+353.jpg,1
+354.jpg,0
+355.jpg,1
+356.jpg,0
+357.jpg,1
+358.jpg,0
+359.jpg,0
+360.jpg,1
+361.jpg,0
+362.jpg,1
+363.jpg,0
+364.jpg,1
+365.jpg,1
+366.jpg,1
+367.jpg,1
+368.jpg,1
+369.jpg,0
+370.jpg,0
+371.jpg,1
+372.jpg,0
+373.jpg,1
+374.jpg,1
+375.jpg,1
+376.jpg,1
+377.jpg,0
+378.jpg,0
+379.jpg,1
+380.jpg,1
+381.jpg,1
+382.jpg,1
+383.jpg,1
+384.jpg,1
+385.jpg,1
+386.jpg,1
+387.jpg,1
+388.jpg,1
+389.jpg,1
+390.jpg,0
+391.jpg,0
+392.jpg,1
+393.jpg,1
+394.jpg,1
+395.jpg,1
+396.jpg,1
+397.jpg,1
+398.jpg,0
+399.jpg,1
+400.jpg,1
+401.jpg,0
+402.jpg,1
+403.jpg,1
+404.jpg,1
+405.jpg,0
+406.jpg,0
+407.jpg,1
+408.jpg,1
+409.jpg,1
+410.jpg,1
+411.jpg,0
+412.jpg,1
+413.jpg,1
+414.jpg,1
+415.jpg,1
+416.jpg,1
+417.jpg,1
+418.jpg,1
+419.jpg,1
+420.jpg,1
+421.jpg,0
+422.jpg,0
+423.jpg,1
+424.jpg,1
+425.jpg,0
+426.jpg,1
+427.jpg,1
+428.jpg,1
+429.jpg,0
+430.jpg,0
+431.jpg,1
+432.jpg,1
+433.jpg,1
+434.jpg,1
+435.jpg,1
+436.jpg,1
+437.jpg,1
+438.jpg,1
+439.jpg,1
+440.jpg,1
+441.jpg,1
+442.jpg,1
+443.jpg,1
+444.jpg,1
+445.jpg,1
+446.jpg,1
+447.jpg,1
+448.jpg,1
+449.jpg,1
+450.jpg,1
+451.jpg,1
+452.jpg,1
+453.jpg,1
+454.jpg,1
+455.jpg,1
+456.jpg,1
+457.jpg,1
+458.jpg,1
+459.jpg,1
+460.jpg,1
+461.jpg,1
+462.jpg,1
+463.jpg,0
+464.jpg,1
+465.jpg,1
+466.jpg,1
+467.jpg,1
+468.jpg,1
+469.jpg,1
+470.jpg,1
+471.jpg,1
+472.jpg,1
+473.jpg,1
+474.jpg,1
+475.jpg,1
+476.jpg,1
+477.jpg,1
+478.jpg,1
+479.jpg,1
+480.jpg,1
+481.jpg,1
+482.jpg,1
+483.jpg,1
+484.jpg,0
+485.jpg,1
+486.jpg,1
+487.jpg,1
+488.jpg,1
+489.jpg,1
+490.jpg,1
+491.jpg,1
+492.jpg,0
+493.jpg,1
+494.jpg,1
+495.jpg,1
+496.jpg,1
+497.jpg,1
+498.jpg,1
+499.jpg,1
+500.jpg,1
+501.jpg,1
+502.jpg,1
+503.jpg,1
+504.jpg,1
+505.jpg,1
+506.jpg,0
+507.jpg,0
+508.jpg,0
+509.jpg,1
+510.jpg,1
+511.jpg,1
+512.jpg,0
+513.jpg,0
+514.jpg,1
+515.jpg,1
+516.jpg,1
+517.jpg,1
+518.jpg,1
+519.jpg,1
+520.jpg,0
--- a/Lab4/dataset/Haze/split_dataset.py
+++ b/Lab4/dataset/Haze/split_dataset.py
@@ -0,0 +1,16 @@
+import os
+import pandas as pd
+import random
+
+train_list = set()
+
+img_list = [i for i in os.listdir("raw/haze") if i.endswith(".jpg")]
+random.shuffle(img_list)
+for img in img_list[ : int(len(img_list) * 0.8)]:
+    train_list.add(img)
+img_list.sort()
+data = list()
+for img in img_list:
+    data.append([img, 1 if img in train_list else 0])
+
+pd.DataFrame(data=data, columns=["Image", "Train"]).to_csv("./split.csv", index=False)