search

JulianJuaner / DeepFloorPlan_Pytorch

A PyTorch implementation for the floor plan segmentation on the r2v dataset. As well as a simple 3D mesh modeling script with the ModelNet dataset.
33 stars 3 forks source link

Pretrain Model #1

Open musdfakoc opened 3 years ago

musdfakoc commented 3 years ago

Hi, I am trying to run your code yet there is no hrnet_w18_small_model_v2.pth file in the repository. So I cannot run it can you provide the pre-trained model too? Thanks already.

FileNotFoundError: [Errno 2] No such file or directory: '/home/user/julianjuaner/research/pretrain/hrnet_w18_small_model_v2.pth'

JulianJuaner commented 2 years ago

sorry for the late reply... you can download pretrained weights from HRNet official repo https://github.com/HRNet/HRNet-Image-Classification

musdfakoc commented 2 years ago

Hi thanks for the return, it solved that issue however, now it gaves: FileNotFoundError: [Errno 2] No such file or directory: './exps/01/ckpt16000.pth' I believe checkpoints are missing in the file. Do you have them or do I have to train it?