How can i get .pkl file conataing bounding boxes with class_id and scores ?

facebookresearch / maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

MIT License

9.31k stars 2.49k forks source link

How can i get .pkl file conataing bounding boxes with class_id and scores ? #1312

Open mathshangw opened 3 years ago

mathshangw commented 3 years ago

I'm new to use MaskRCNN i tried to train the model on my dataset and got 1000 epochs should be finished .. Will train phase give me checkpoints.pth file ? Should i use this file for getting .pkl ? or is there another way i can use ?