| 101 | main_layout.addWidget(right_panel, 1) |
| 102 | |
| 103 | def create_left_panel(self): |
| 104 | panel = QWidget() |
| 105 | layout = QVBoxLayout() |
| 106 | panel.setLayout(layout) |
| 107 | |
| 108 | title = QLabel('🚀 NeuralForge Model Tester') |
| 109 | title.setFont(QFont('Arial', 20, QFont.Weight.Bold)) |
| 110 | title.setAlignment(Qt.AlignmentFlag.AlignCenter) |
| 111 | layout.addWidget(title) |
| 112 | |
| 113 | model_group = QGroupBox('Model Selection') |
| 114 | model_layout = QVBoxLayout() |
| 115 | |
| 116 | model_path_layout = QHBoxLayout() |
| 117 | self.model_path_input = QLineEdit() |
| 118 | self.model_path_input.setPlaceholderText('Path to model file (.pt)') |
| 119 | model_path_layout.addWidget(self.model_path_input) |
| 120 | |
| 121 | browse_btn = QPushButton('Browse') |
| 122 | browse_btn.clicked.connect(self.browse_model) |
| 123 | model_path_layout.addWidget(browse_btn) |
| 124 | |
| 125 | default_btn = QPushButton('Use Default') |
| 126 | default_btn.clicked.connect(self.use_default_model) |
| 127 | model_path_layout.addWidget(default_btn) |
| 128 | |
| 129 | model_layout.addLayout(model_path_layout) |
| 130 | |
| 131 | dataset_layout = QHBoxLayout() |
| 132 | dataset_label = QLabel('Dataset:') |
| 133 | self.dataset_input = QLineEdit('cifar10') |
| 134 | self.dataset_input.setPlaceholderText('cifar10, mnist, stl10, tiny_imagenet, etc.') |
| 135 | self.dataset_input.setToolTip('Supported: cifar10, cifar100, mnist, fashion_mnist, stl10,\ntiny_imagenet, imagenet, food101, caltech256, oxford_pets') |
| 136 | dataset_layout.addWidget(dataset_label) |
| 137 | dataset_layout.addWidget(self.dataset_input) |
| 138 | model_layout.addLayout(dataset_layout) |
| 139 | |
| 140 | self.load_model_btn = QPushButton('Load Model') |
| 141 | self.load_model_btn.clicked.connect(self.load_model) |
| 142 | model_layout.addWidget(self.load_model_btn) |
| 143 | |
| 144 | self.model_status = QLabel('No model loaded') |
| 145 | self.model_status.setAlignment(Qt.AlignmentFlag.AlignCenter) |
| 146 | model_layout.addWidget(self.model_status) |
| 147 | |
| 148 | model_group.setLayout(model_layout) |
| 149 | layout.addWidget(model_group) |
| 150 | |
| 151 | image_group = QGroupBox('Image Selection') |
| 152 | image_layout = QVBoxLayout() |
| 153 | |
| 154 | image_path_layout = QHBoxLayout() |
| 155 | self.image_path_input = QLineEdit() |
| 156 | self.image_path_input.setPlaceholderText('Path to image file') |
| 157 | image_path_layout.addWidget(self.image_path_input) |
| 158 | |
| 159 | browse_image_btn = QPushButton('Browse') |
| 160 | browse_image_btn.clicked.connect(self.browse_image) |