DmitriyKras / Small-objects-segmentation

This repository contains project about segmentation small flying objects with U-Net, PSP-Net and FCN
4 stars 0 forks source link

Can you provide training models for testing ? #1

Open assertdebug opened 1 month ago

assertdebug commented 1 month ago

Can you provide training models for testing ?

DmitriyKras commented 1 month ago

Hello! Unfortunately, I can't. This repo is my master thesis work so I've removed all datasets and weights from my laptop after graduation(( I'll try to search in Google Drive, perhaps I might find something

assertdebug commented 1 month ago

Which one was used for the pre-trained model?

DmitriyKras commented 1 month ago

I've searched my Google Drive and unfortunately couldn't find these weights. All models presented in this repo were trained from scratch. So there wasn't any pre-trained model. If it helps, pictures with prediction examples were made with trained UNet because it showed the best results among other models.