[Update]update for zhongzaicanyu of IJCAI_2024 (#971)

* [Update]update for zhongzaicanyu of IJCAI_2024

* update README.txt

Author: lijialin03

jointContribution/IJCAI_2024/README.md

@@ -41,7 +41,7 @@ mv ./ckpts/bju/geom/ckpt ./bju/geom/
 mv ./ckpts/bju/pretrained_checkpoint.pdparams ./bju/pretrained_checkpoint.pdparams
 
 # zhongzaicanyu
-# No pretrained checkpoint yet.
+mv ./ckpts/zhongzaicanyu/pretrained_checkpoint.pdparams ./zhongzaicanyu/pretrained_checkpoint.pdparams
 ```
 
 ## Inference
@@ -74,5 +74,5 @@ python infer.py
 python infer.py --train_data_dir "./Dataset/Trainset_track_B" --test_data_dir "./Dataset/Testset_track_B/Inference" --info_dir "./Dataset/Testset_track_B/Auxiliary" --ulip_ckpt "./geom/ckpt/checkpoint_pointbert.pdparams"
 
 ### zhongzaicanyu
-python infer.py # not work yet.
+python infer.py --data_dir "./Dataset/data_centroid_track_B_vtk" --test_data_dir "./Dataset/track_B_vtk" --save_dir "./Dataset/data_centroid_track_B_vtk_preprocessed_data"
 ```

jointContribution/IJCAI_2024/zhongzaicanyu/dataset.py

@@ -457,6 +457,8 @@ def __getitem__(self, idx):
     def collate_fn(self, batch):
         batch_data = [data for (data, _) in batch]
         batch_shape = paddle.stack([shape for (_, shape) in batch], axis=0)
+        if len(batch_data) == 1:
+            return batch_data[0], batch_shape
         return batch_data, batch_shape
 
 

jointContribution/IJCAI_2024/zhongzaicanyu/download_dataset.ipynb

@@ -54,7 +54,10 @@ def parse_args():
 
 
 if __name__ == "__main__":
-    # only run it first time
+    print(
+        "Attention: Please run and only run `data_process()` at first time in `infer.py`. "
+        "And change path in the file before run it."
+    )
     data_process()
 
     # load setting
@@ -126,8 +129,6 @@ def parse_args():
                 print(f"Processing mesh index: {mesh_index}")
             else:
                 raise ValueError(f"Invalid mesh file format: {mesh_file}")
-            cfd_data = cfd_data.to(device)
-            geom = geom.to(device)
             tic = time.time()
             out = model((cfd_data, geom))
             toc = time.time()
@@ -138,12 +139,7 @@ def parse_args():
                 press_output = press_output * std_out[-1] + mean_out[-1]
             press_output = press_output.detach().cpu().numpy()
             np.save(
-                "./results/"
-                + args.cfd_model
-                + "_B/"
-                + "press"
-                + "_"
-                + f"{mesh_index}.npy",
+                "./results/" + "press" + "_" + f"{mesh_index}.npy",
                 press_output,
             )
             times.append(toc - tic)

jointContribution/IJCAI_2024/zhongzaicanyu/infer.py

@@ -0,0 +1,6 @@
+einops
+numpy
+paddlepaddle_gpu
+scikit_learn
+tqdm
+vtk