how to make csv file code example
Example: Creating csv file
import pandas as pdimport osimport torchdevice = ("cuda" if torch.cuda.is_available() else "cpu")train_df = pd.DataFrame(columns=["img_name","label"])train_df["img_name"] = os.listdir("train/")for idx, i in enumerate(os.listdir("train/")): if "cat" in i: train_df["label"][idx] = 0 if "dog" in i: train_df["label"][idx] = 1train_df.to_csv (r'train_csv.csv', index = False, header=True)