This commit is contained in:
Eric 2021-03-16 17:29:32 -04:00
parent 4df7ba11bf
commit 6a1e031846
2 changed files with 3 additions and 1 deletions

View File

@ -14,12 +14,12 @@
* limitations under the License.
*/
#include <nlohmann/json.hpp> // we need this before the include to avoid ambiguity in nullptr conversion
#include "minddata/dataset/include/datasets.h"
#include <algorithm>
#include <fstream>
#include <unordered_set>
#include <utility>
#include <nlohmann/json.hpp> // we need this before the include to avoid ambiguity in nullptr conversion
#include "minddata/dataset/core/tensor.h"
#include "minddata/dataset/engine/runtime_context.h"

View File

@ -1126,6 +1126,8 @@ class RandomRotation(ImageTensorOperation):
self.fill_value = fill_value
def parse(self):
# pylint false positive
# pylint: disable=E1130
degrees = (-self.degrees, self.degrees) if isinstance(self.degrees, numbers.Number) else self.degrees
interpolation = DE_C_INTER_MODE[self.resample]
expand = self.expand