araujoalexandre / focusstackingdataset Goto Github PK

Thanks for your work! I meet some problem when I try to run your code:
In Class FocusDataset, something went wrong with "crop_path = f'{burst_path}/crops{self.dataset_id}/crop{i}.pkl'". I searched the dataset folder but did not find such files named ".pkl", and I did not find the code to generate these ".pkl" files. Could you tell me how to generate these files? Thank you!

How to use your code on a costume dataset? I have a set 60 raw images and I want to stack them. Do I need to train the model with a stacked image by Helicon Focus first and then use it? Or There is a pre-trained model that can be applied to arbitrary images?