Description: This repo is actually forked from virajmavani and implemented for the pytorch models. Labelling of image can be done easily with the help of pytorch pretrained model of the trained clases
- Add the pytorch model in the snapshot/pytorch folder
- And the name of the model must be in the architecture name
- eg: yolo.pt , resnet32.pt etc..
run main.py
- An UI labellig will be display
- Select the model and add it , then use detect to use the semi annotation tool