| 1234567891011121314151617181920212223 |
- import torch
- from models.line_detect.line_detect import linedetect_newresnet18fpn, linedetect_resnet50_fpn, linedetect_resnet18_fpn, \
- linedetect_newresnet50fpn
- from models.line_net.trainer import Trainer
- device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
- if __name__ == '__main__':
- # model = LineNet('line_net.yaml')
- # model=linenet_resnet50_fpn()
- # model = linedetect_resnet50_fpn()
- # model=get_line_net_convnext_fpn(num_classes=2).to(device)
- # model=linenet_newresnet50fpn()
- # model = lineDetect_resnet18_fpn()
- # model=linedetect_resnet18_fpn()
- model=linedetect_newresnet50fpn(num_points=3)
- # model = linedetect_newresnet50fpn(num_points=3)
- # model.load_weights(save_path=r'/home/admin/projects/MultiVisionModels/models/line_detect/train_results/20250711_114046/weights/best_val.pth')
- model.start_train(cfg='train.yaml')
|
