Test for my custom file

Hi, first, thanks for sharing more easier and custom version of FeatureNet.

I'm trying to use FeatureNet for my custom CAD file, and i am just a beginner for a coding or something elses. The purpose of this is just tutorial about FeatureNet. I finished training and testing the model with dataset, and now i'm trying to test model with my custom CAD file.

What i understand to test it is 1. Voxelization and make h5 file, 2. model.predict("predict_file.h5")

I used stl_to_voxel.py to convert my file to voxel and h5 file, the dim of my file is 3, so i should expand my voxel data to 5.(Because model wants 5 dims input data.) Which shape should my data be? (ex, (1, 64, 64, 64, 1)? (1, 1, 64, 64, 64)?) I'm really confused. Or did i misunderstand the step?