Skip to content
Snippets Groups Projects
Commit 86d30d7a authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!945 C3d preprocess bug fix

Merge pull request !945 from 张晓晓/master
parents 346af715 105b4eb5
No related branches found
No related tags found
No related merge requests found
......@@ -32,6 +32,8 @@ def gen_bin(data_dir):
image_path = os.path.join(data_dir, "image")
label_path = os.path.join(data_dir, "label_bs" + str(config.batch_size) + ".npy")
label_list = []
if not os.path.isdir(image_path):
os.makedirs(image_path)
for index, (input_data, label) in enumerate(dataset):
f_name = "c3d_bs" + str(config.batch_size) + "_" + str(index) + ".bin"
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment