saving_model

from icevision.all import *

classes = ['Army_navy', 'Bulldog', 'Castroviejo','Forceps', 'Frazier', 'Hemostat','Iris','Mayo_metz','Needle','Potts','Richardson','Scalpel','Towel_clip', 'Weitlaner','Yankauer']
class_map = ClassMap(classes)

from icevision.models import *

checkpoint_path = root_dir/'Surgical_instruments/Models/Model_M_set11_ret_nov8_map63.6.pth' #100 epochs, vf

save_icevision_checkpoint(model, 
                        model_name='mmdet.retinanet', ## 
                        backbone_name='resnet50_fpn_1x', ##
                        classes =  train_parser.class_map.get_classes(), 
                        img_size=image_size, 
                        filename=str(checkpoint_path), ##
                        meta={'icevision_version': '0.9.1'})
                        
checkpoint_and_model = model_from_checkpoint(checkpoint_path)

model_loaded = checkpoint_and_model["model"]