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):
87
87
self .assertEqual (img ["annotationfile" ]["classification_label" ], "no-corrosion" )
88
88
89
89
def test_parse_multilabel_classification_csv (self ):
90
- folder = f"{ thisdir } /../datasets/multilabel-classification"
90
+ folder = f"{ thisdir } /../datasets/skinproblem- multilabel-classification"
91
91
parsed = folderparser .parsefolder (folder , is_classification = True )
92
92
images = {img ["name" ]: img for img in parsed ["images" ]}
93
- img1 = images .get ("img1 .jpg" )
93
+ img1 = images .get ("101_jpg.rf.ffb91e580c891eb04b715545274b2469 .jpg" )
94
94
self .assertIsNotNone (img1 )
95
95
self .assertEqual (img1 ["annotationfile" ]["type" ], "classification_multilabel" )
96
- self .assertEqual (set (img1 ["annotationfile" ]["labels" ]), {"rust " })
96
+ self .assertEqual (set (img1 ["annotationfile" ]["labels" ]), {"Blackheads " })
97
97
98
98
99
99
def _assertJsonMatchesFile (actual , filename ):
You can’t perform that action at this time.
0 commit comments