buildingnet / buildingnet_dataset

https://buildingnet.org/
56 stars 17 forks source link

Model verification error: `FileNotFoundError: [Errno 2] No such file or directory: 'data/pretrained_avgpool_minkownormal_features/fc3_avg/RESIDENTIALhouse_mesh1367.pth.tar'` #16

Closed ozanyetkin closed 8 months ago

ozanyetkin commented 9 months ago

Hello and thank you for sharing your great work!

I have faced an issue about the following:

Problem

When the BuildingNet dataset is downloaded by filling out the form, there are several versions available, yet it seems like some models are missing in pretrained_avgpool_minkownormal_features.

Example

Running the following code as BuildingNet repository suggests:

python3 train.py --datadir="data" --epoch 200 --outputdir 'Output' --ckpt_dir 'checkpoint' --normalization 'GN' --modeltype 'Edge' --edgetype 'all' --lr 1e-4 --nodetype 'node+minkownormal' --pretrainedtype 'fc3_avg' --IOU_checkpoint=5

Throws the following error:

FileNotFoundError: [Errno 2] No such file or directory: 'data/pretrained_avgpool_minkownormal_features/fc3_avg/RESIDENTIALhouse_mesh1367.pth.tar'

Yet, I have resolved it with a simple script and shared it via this repository.

In case anyone faces the same issue, feel free to use it!

buildingnet commented 9 months ago

Presently, only BuildingNet v0 offers pretrained MinkNet features, which have been trained on the entirety of BuildingNet v0, covering 2,000 building models. For BuildingNet v1, which comprises a subset of v0 containing a total of 1,849 building models, you can access the pretrained MinkNet features (trained specifically on BuildingNet v1) by following the instructions outlined in this guide.

ozanyetkin commented 9 months ago

Great, thanks! I will try to extend the dataset for my ongoing research. I will probably share the updates from my forked version of BuildingNet.