File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -87,13 +87,13 @@ def test_parse_classification_folder_structure(self):
8787 self .assertEqual (img ["annotationfile" ]["classification_label" ], "no-corrosion" )
8888
8989 def test_parse_multilabel_classification_csv (self ):
90- folder = f"{ thisdir } /../datasets/multilabel-classification"
90+ folder = f"{ thisdir } /../datasets/skinproblem- multilabel-classification"
9191 parsed = folderparser .parsefolder (folder , is_classification = True )
9292 images = {img ["name" ]: img for img in parsed ["images" ]}
93- img1 = images .get ("img1 .jpg" )
93+ img1 = images .get ("101_jpg.rf.ffb91e580c891eb04b715545274b2469 .jpg" )
9494 self .assertIsNotNone (img1 )
9595 self .assertEqual (img1 ["annotationfile" ]["type" ], "classification_multilabel" )
96- self .assertEqual (set (img1 ["annotationfile" ]["labels" ]), {"rust " })
96+ self .assertEqual (set (img1 ["annotationfile" ]["labels" ]), {"Blackheads " })
9797
9898
9999def _assertJsonMatchesFile (actual , filename ):
You can’t perform that action at this time.
0 commit comments